Update app.py

app.py

@@ -8,11 +8,14 @@ model_path = MODEL_PATHS["YOLOv8n"]
 model = YOLO(model_path)
 
 # ---------- Detection Functions ----------
+
 def detect_image_fn(image, confidence):
     if image is None:
         return None
-    output_paths = detect_from_images([image], model, confidence)
-    return output_paths[0] if output_paths else None
+    image_np = np.array(image)  # Convert PIL to NumPy
+    results = model.predict(image_np, conf=confidence, verbose=False)
+    annotated = results[0].plot()
+    return annotated
 
 def detect_video_fn(video_file, confidence):
     if video_file is None: