Update app.py

app.py

@@ -4,30 +4,32 @@ from PIL import Image
 import numpy as np
 import sys, os
 
-# ---- Add forked YOLOv12 code to Python path ----
+# ---- Add YOLOv12 code to Python path ----
 fork_ultra_path = os.path.join(os.path.dirname(__file__), "yolov12-main", "yolov12-main")
 sys.path.insert(0, fork_ultra_path)
 
-# ---- Now import YOLO from the fork, not from pip ----
+# ---- Import YOLO ----
 from ultralytics import YOLO
 
-# ---- Load the trained model ----
-model = YOLO("best.pt")
+# ---- Load trained model ----
+model_path = os.path.join(os.path.dirname(__file__), "best.pt")
+model = YOLO(model_path)
 
-# ---- Inference function ----
+# ---- Detection function ----
 def detect_objects(image):
     img = np.array(image)
     results = model(img)
     annotated = results[0].plot()
     return Image.fromarray(annotated)
 
-# ---- Gradio UI ----
-iface = gr.Interface(
+# ---- Gradio Interface ----
+demo = gr.Interface(
     fn=detect_objects,
-    inputs=gr.Image(type="pil"),
-    outputs=gr.Image(type="pil"),
-    title="Ear Condition Detection (YOLOv12 Fork)",
-    description="Runs inference using the sunsmarterjie YOLOv12 fork."
+    inputs=gr.Image(type="pil", label="Upload an Ear Image"),
+    outputs=gr.Image(type="pil", label="Detection Result"),
+    title="Ear Condition Detection",
+    description="Upload an ear image to detect possible conditions using the trained YOLOv12 model."
 )
 
-iface.launch()
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)