Update app.py

app.py

@@ -10,13 +10,15 @@ import shutil
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
 
 def detect_video(video):
-    # Create a temporary file to save the video
-    with tempfile.NamedTemporaryFile(delete=False) as tmpfile:
-        tmpfile.write(video.read())  # Save the uploaded video to the temp file
-        video_path = tmpfile.name  # Get the temporary file path
-    
-    cap = cv2.VideoCapture(video_path)  # Open the temporary video file
+    # Check if the video is in bytes (Gradio typically passes it as bytes)
+    if isinstance(video, bytes):
+        with tempfile.NamedTemporaryFile(delete=False) as tmpfile:
+            tmpfile.write(video)  # Write the bytes to the temp file
+            video_path = tmpfile.name  # Get the temp file path
+    else:
+        video_path = video  # In case it's already a file path
 
+    cap = cv2.VideoCapture(video_path)  # Open the temporary video file
 
     # List to hold results
     detection_results = []