Update app.py

app.py

@@ -5,6 +5,7 @@ import numpy as np
 import tempfile
 from datetime import datetime
 import shutil
+import os  # Import os module for file removal
 
 # Load YOLOv5 model
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
@@ -45,8 +46,8 @@ def detect_video(video):
     
     cap.release()
     
-    # Clean up the temporary file
-    shutil.remove(video_path)
+    # Clean up the temporary file using os.remove
+    os.remove(video_path)
 
     return detection_results