Update app.py

app.py

@@ -1,37 +1,35 @@
 import gradio as gr
-import numpy as np
 from PIL import Image
-from ultralytics import YOLO
-from pathlib import Path
+import sys
+import os
 
-# ✅ Load YOLOv12 model (make sure this path matches your repo)
-model_path = Path("best.pt")  # Or change to "weights/best.pt" if uploaded in subfolder
-assert model_path.exists(), f"Model not found at {model_path}"
-model = YOLO(str(model_path))
+# 👇 Add your custom YOLOv12 path to Python path
+sys.path.append("yolov12")
+
+# 👇 Import your custom model loader
+from models.yolo import YOLO  # adjust based on your repo
+
+model = YOLO("yolov12/best.pt")
 
-# ✅ Damage detection using YOLOv12
 def detect_damage(image: Image.Image):
     results = model(image)
-    
     is_damaged = False
+
     for result in results:
         for box in result.boxes:
             class_id = int(box.cls[0])
             label = model.names[class_id]
-            print("Detected label:", label)
+            print("Detected:", label)
             if "apple_damaged" in label.lower():
                 is_damaged = True
                 break
 
     return {"is_damaged": is_damaged}
 
-# ✅ Gradio interface
-demo = gr.Interface(
-    fn=detect_damage,
-    inputs=gr.Image(type="pil"),
-    outputs=gr.JSON(label="Detection Result"),
-    title="🍎 Apple Damage Detector",
-    description="Upload an apple image to detect whether it is damaged using a YOLOv12 model."
-)
+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.launch()