Update app.py

app.py

@@ -1,15 +1,15 @@
 import gradio as gr
 from PIL import Image
 import sys
-import os
 
-# 👇 Add your custom YOLOv12 path to Python path
+# Add yolov12 to Python path
 sys.path.append("yolov12")
 
-# 👇 Import your custom model loader
-from models.yolo import YOLO  # adjust based on your repo
+# Correct import from yolov12 repo
+from yolov12.models.yolo import YOLO
 
-model = YOLO("yolov12/best.pt")
+# ✅ Correct path to the actual weights file
+model = YOLO("yolov12/runs/detect/train7/weights/best.pt")
 
 def detect_damage(image: Image.Image):
     results = model(image)
@@ -26,10 +26,12 @@ def detect_damage(image: Image.Image):
 
     return {"is_damaged": is_damaged}
 
-demo = gr.Interface(fn=detect_damage,
-                    inputs=gr.Image(type="pil"),
-                    outputs=gr.JSON(label="Detection Result"),
-                    title="🍎 Apple Damage Detector (YOLOv12)",
-                    description="Upload an image of an apple to check if it's damaged.")
+demo = gr.Interface(
+    fn=detect_damage,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.JSON(label="Detection Result"),
+    title="🍎 Apple Damage Detector (YOLOv12)",
+    description="Upload an image of an apple to check if it's damaged using your custom YOLOv12 model."
+)
 
 demo.launch()