added simple progress bar

app_batch.py

@@ -25,7 +25,8 @@ def run_owl_batch(
     species_prompt: str,
     conf_threshold: float,
     fps_processed: int,
-    scaling_factor: float
+    scaling_factor: float, 
+    progress=gr.Progress()
 ) -> tuple[str, str, str]:
     """
     args: 
@@ -65,7 +66,8 @@ def run_owl_batch(
                     fps_processed=fps_processed, 
                     scaling_factor=1/scaling_factor, 
                     batch_size=BATCH_SIZE, 
-                    save_dir=f"temp_{timestamp}")
+                    save_dir=f"temp_{timestamp}", 
+                    progress=progress)
 
     end_time = time.time()
     print(f'Processing time: {end_time - start_time} seconds')

owl_batch.py

@@ -2,6 +2,7 @@ import os
 import shutil
 from tqdm import tqdm
 import cv2
+import gradio as gr
 import pandas as pd
 import torch
 from PIL import Image
@@ -18,14 +19,15 @@ def owl_batch_video(
     fps_processed: int = 1,
     scaling_factor: float = 0.5, 
     batch_size: int = 8, 
-    save_dir: str = "temp/"
+    save_dir: str = "temp/", 
+    progress=gr.Progress()
 ):
     pos_preds = []
     neg_preds = []
 
     df = pd.DataFrame(columns=["video path", "detection?"])
 
-    for vid in input_vids:
+    for vid in progress.tqdm(input_vids, desc="Processing videos"):
         detection = owl_video_detection(vid, 
                             target_prompt, 
                             species_prompt,