Update app.py

app.py

@@ -1,15 +1,12 @@
+import os
+import numpy as np
+from PIL import Image, ImageDraw
 import gradio as gr
 from ultralytics import YOLO
-from PIL import Image
-import numpy as np
-import os
 
 # --- Load YOLOv12 model ---
 model_path = "best.pt"
-if os.path.exists(model_path):
-    model = YOLO(model_path)
-else:
-    model = None
+model = YOLO(model_path) if os.path.exists(model_path) else None
 
 # --- Inference function ---
 def detect_ear(image: Image.Image, conf: float = 0.25):
@@ -21,9 +18,8 @@ def detect_ear(image: Image.Image, conf: float = 0.25):
             return Image.fromarray(annotated)
         except Exception as e:
             print(f"Model inference failed: {e}")
-    
+
     # --- Dummy fallback ---
-    import PIL.ImageDraw as ImageDraw
     im = image.copy()
     draw = ImageDraw.Draw(im)
     w, h = im.size
@@ -41,7 +37,7 @@ iface = gr.Interface(
     ],
     outputs=gr.Image(type="pil", label="Detection Result"),
     title="Automated Ear Disease Detection",
-    description="Detect ear conditions using YOLOv12",
+    description="Detect ear conditions using YOLOv12. If the model fails to load, a dummy detection will be shown."
 )
 
 # --- Launch App ---