Update app.py

app.py

@@ -1,20 +1,21 @@
 import gradio as gr
 from PIL import Image
+import os
 import sys
 
-# Add yolov12 to Python path
-sys.path.append("yolov12")
+# Make yolov12 folder importable
+sys.path.append(os.path.abspath("yolov12"))
 
-# Correct import from yolov12 repo
-from yolov12.models.yolo import YOLO
+# Import your YOLO class
+from yolo import YOLO  # Now from yolov12/yolo.py
 
-# ✅ Correct path to the actual weights file
-model = YOLO("yolov12/runs/detect/train7/weights/best.pt")
+# Load your custom YOLOv12 model
+model = YOLO("yolov12/best.pt")
 
 def detect_damage(image: Image.Image):
     results = model(image)
-    is_damaged = False
 
+    is_damaged = False
     for result in results:
         for box in result.boxes:
             class_id = int(box.cls[0])
@@ -31,7 +32,7 @@ demo = gr.Interface(
     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."
+    description="Upload an image of an apple to detect if it's damaged."
 )
 
 demo.launch()