Update app.py

app.py

@@ -8,11 +8,11 @@ model_path = MODEL_PATHS["YOLOv8n"]
 model = YOLO(model_path)
 
 # ---------- Detection Functions ----------
-def detect_image_fn(image_path, confidence):
-    if image_path is None:
+def detect_image_fn(image, confidence):
+    if image is None:
         return None
-    results = detect_from_images([image_path], model, confidence, return_path=True)
-    return results[0] if results else None
+    output_paths = detect_from_images([image], model, confidence, return_path=True)
+    return output_paths[0] if output_paths else None
 
 def detect_video_fn(video_file, confidence):
     if video_file is None:
@@ -27,7 +27,7 @@ def detect_video_fn(video_file, confidence):
 image_interface = gr.Interface(
     fn=detect_image_fn,
     inputs=[
-        gr.Image(label="Upload Image"),  # No type="filepath"
+        gr.Image(type="filepath", label="Upload Image"),
         gr.Slider(0.1, 1.0, value=0.3, step=0.05, label="Confidence")
     ],
     outputs=gr.Image(label="Predicted Output"),
@@ -38,7 +38,7 @@ image_interface = gr.Interface(
 video_interface = gr.Interface(
     fn=detect_video_fn,
     inputs=[
-        gr.Video(label="Upload Video"),  # No type="file"
+        gr.Video(type="filepath", label="Upload Video"),
         gr.Slider(0.1, 1.0, value=0.3, step=0.05, label="Confidence")
     ],
     outputs=gr.Video(label="Predicted Output"),
@@ -50,5 +50,5 @@ video_interface = gr.Interface(
 gr.TabbedInterface(
     [image_interface, video_interface],
     tab_names=["Image Detection", "Video Detection"]
-).launch(share=True)
+).launch()