Update app.py

app.py

@@ -1,12 +1,17 @@
 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_path):
-    cap = cv2.VideoCapture(video_path)
+def object_detection(video_data):
+    # Convert video data to numpy array
+    video_bytes = video_data.read()
+    video_np = np.frombuffer(video_bytes, np.uint8)
+    
+    cap = cv2.VideoCapture(video_np)
     
     while True:
         ret, frame = cap.read()