Update app.py

app.py

@@ -2,14 +2,20 @@ import torch
 import gradio as gr
 import cv2
 import numpy as np
+import tempfile
 from datetime import datetime
+import shutil
 
 # Load YOLOv5 model
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
 
 def detect_video(video):
-    # Use the video file provided by Gradio
-    cap = cv2.VideoCapture(video.name)  # video.name gives the path to the temporary file
+    # 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
 
     # List to hold results
     detection_results = []
@@ -36,6 +42,9 @@ def detect_video(video):
     
     cap.release()
     
+    # Clean up the temporary file
+    shutil.remove(video_path)
+
     return detection_results
 
 # Gradio Interface