Update app.py

app.py

@@ -17,9 +17,19 @@ model = YOLO(model_path)
 
 # ---- Detection function ----
 def detect_objects(image):
-    img = np.array(image)
-    # use predict() with confidence and image size settings
-    results = model.predict(source=img, imgsz=640, conf=0.25)
+    # Convert PIL to RGB NumPy array (YOLO expects RGB)
+    img = np.array(image.convert("RGB"))
+
+    # Run prediction with same settings used during training
+    results = model.predict(
+        source=img,
+        imgsz=640,        # same resolution as training
+        conf=0.25,        # lower confidence threshold to catch subtle detections
+        iou=0.45,         # standard non-max suppression
+        verbose=False
+    )
+
+    # Visualize results
     annotated = results[0].plot()
     return Image.fromarray(annotated)