handle stopiteration

app.py

@@ -57,7 +57,10 @@ def process_video(
     all_fps = []
     with sv.VideoSink(result_file_path, video_info=video_info) as sink:
         for _ in tqdm(range(total), desc="Processing video.."):
-            frame = next(frame_generator)
+            try:
+                frame = next(frame_generator)
+            except StopIteration:
+                break
             results, fps = query(
                 frame, classes, confidence_threshold, max_side=max_side
             )