Update app.py

app.py

@@ -1,17 +1,23 @@
 from ultralytics import YOLO
 import gradio as gr
 from PIL import Image
+import numpy as np
 
-# Load your trained YOLOv12 model from the repo
-model = YOLO("best.pt")  # just the file name if it's in the same folder as app.py
+# Load YOLOv12 model
+model = YOLO("best.pt")  # model in repo
 
-# Define prediction function
 def predict(image):
-    results = model.predict(image)  # Run inference
-    annotated_img = results[0].plot()  # Draw boxes on image
+    # Ensure RGB and convert to NumPy array
+    image = image.convert("RGB")
+    img_array = np.array(image)
+    
+    # Run inference
+    results = model.predict(img_array)
+    
+    # Annotate and return
+    annotated_img = results[0].plot()
     return annotated_img
 
-# Gradio interface
 iface = gr.Interface(
     fn=predict,
     inputs=gr.Image(type="pil"),
@@ -20,5 +26,4 @@ iface = gr.Interface(
     description="Upload an otoscopic image and get predictions from the trained YOLOv12 model."
 )
 
-# Launch the app
 iface.launch()