Update app.py

app.py

@@ -1,20 +1,17 @@
 import gradio as gr
 import torch
-from ultralytics import YOLO
 from PIL import Image
 import numpy as np
 import sys, os
 
-# ---- Add yolov12 fork folder to path ----
-sys.path.append(os.path.join(os.path.dirname(__file__), "yolov12-main"))
+# ---- Add forked YOLOv12 code to Python path ----
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), "yolov12-main"))
 
-# ---- Import forked Attn and patch YOLO ----
-from yolov12.models.modules.attn import Attn  # inside yolov12-main
-import ultralytics.nn.modules.block as block
-block.AAttn = Attn  # replace original AAttn
+# ---- Now import YOLO from the fork, not from pip ----
+from ultralytics import YOLO
 
-# ---- Load model ----
-model = YOLO("best.pt")  # make sure best.pt is in same folder
+# ---- Load the trained model ----
+model = YOLO("best.pt")
 
 # ---- Inference function ----
 def detect_objects(image):
@@ -23,13 +20,13 @@ def detect_objects(image):
     annotated = results[0].plot()
     return Image.fromarray(annotated)
 
-# ---- Gradio interface ----
+# ---- Gradio UI ----
 iface = gr.Interface(
     fn=detect_objects,
     inputs=gr.Image(type="pil"),
     outputs=gr.Image(type="pil"),
     title="Ear Condition Detection (YOLOv12 Fork)",
-    description="Uses forked YOLOv12 model to detect ear conditions."
+    description="Runs inference using the sunsmarterjie YOLOv12 fork."
 )
 
 iface.launch()