Update app.py

app.py

@@ -26,6 +26,8 @@ def process_video(video):
     # List to store processed frames for Gradio output
     processed_frames = []
 
+    frame_counter = 0  # Counter to limit the number of frames for processing
+
     while True:
         # Read a frame from the video
         ret, frame = input_video.read()
@@ -49,6 +51,11 @@ def process_video(video):
             # Append the annotated frame to the list for Gradio
             processed_frames.append(annotated_frame_rgb)
 
+        # Limit the number of frames processed (to speed up processing)
+        frame_counter += 1
+        if frame_counter > 100:  # Process only 100 frames, adjust as necessary
+            break
+
     # Release resources
     input_video.release()