Update app.py

app.py

@@ -64,15 +64,23 @@ def process_video(video_path, target, progress=gr.Progress()):
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(output_path, fourcc, original_fps, (int(cap.get(3)), int(cap.get(4))))
 
+    batch_size = 16
+    frames = []
+
     for frame in progress.tqdm(range(frame_count)):
         ret, img = cap.read()
         if not ret:
             break
 
         pil_img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
-        annotated_img = detect_objects_in_frame(pil_img, target)
-        annotated_frame = cv2.cvtColor(np.array(annotated_img), cv2.COLOR_RGB2BGR)
-        out.write(annotated_frame)
+        frames.append(pil_img)
+
+        if len(frames) == batch_size or frame == frame_count - 1:
+            annotated_frames = [detect_objects_in_frame(frame, target) for frame in frames]
+            for annotated_img in annotated_frames:
+                annotated_frame = cv2.cvtColor(np.array(annotated_img), cv2.COLOR_RGB2BGR)
+                out.write(annotated_frame)
+            frames = []
 
     cap.release()
     out.release()