Update app.py

app.py

@@ -1,13 +1,12 @@
 import gradio as gr
 import cv2
 import torch
-import numpy as np
 
 # Load YOLOv5 model
 model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True)
 
-def object_detection(video):
-    cap = cv2.VideoCapture(video.name)
+def object_detection(video_path):
+    cap = cv2.VideoCapture(video_path)
     
     while True:
         ret, frame = cap.read()
@@ -28,5 +27,5 @@ def object_detection(video):
     
     cap.release()
 
-iface = gr.Interface(fn=object_detection, inputs=gr.Video(), outputs=gr.Image(label="Object Detection"))
+iface = gr.Interface(fn=object_detection, inputs=gr.Video(label="Upload Video"), outputs=gr.Image(label="Object Detection"))
 iface.launch()