Update video_detection.py

video_detection.py

@@ -31,6 +31,7 @@ def draw_boxes(frame, results):
 
     return frame
     
+@spaces.GPU   
 def video_detection(cap):
     video_codec = cv2.VideoWriter_fourcc(*"mp4v") # type: ignore
     #video_codec = cv2.VideoWriter_fourcc(*'h264')
@@ -53,7 +54,7 @@ def video_detection(cap):
         if n_frames % SUBSAMPLE == 0:
             batch.append(frame)
         #if len(batch) == 2 * desired_fps:
-        if len(batch) == 8:
+        if len(batch) == 4:
             #inputs = image_processor(images=batch, return_tensors="pt").to("cuda")
 
             print(f"starting batch of size {len(batch)}")