Update app.py

app.py

@@ -34,7 +34,26 @@ def detect_video(model_choice, input_video=None, sample_video_choice=None):
     model = load_model(models[model_choice])
     if sample_video_choice:
         input_video = sample_videos[sample_video_choice]
-    return model.detect_video(input_video)
+
+    # Resize the video to 640x640 for faster processing
+    cap = cv2.VideoCapture(input_video)
+    output_path = "output_video.mp4"
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    out = cv2.VideoWriter(output_path, fourcc, 20.0, (640, 640))
+
+    while cap.isOpened():
+        ret, frame = cap.read()
+        if not ret:
+            break
+        frame = cv2.resize(frame, (640, 640))
+        # Detect objects in the frame
+        detection_result = model.detect_video(frame)
+        out.write(detection_result)
+
+    cap.release()
+    out.release()
+    
+    return output_path
 
 app = gr.Blocks()