remove save folder for storage

app_batch.py

@@ -12,6 +12,7 @@ import gradio as gr
 from tqdm import tqdm
 import os
 import time
+import shutil
 
 from owl_batch import owl_batch_video
 
@@ -58,6 +59,8 @@ def run_owl_batch(
     now = datetime.now()
     timestamp = now.strftime("%Y-%m-%d_%H-%M")
 
+    save_dir = f"temp_{timestamp}"
+
     zip_path = owl_batch_video(
                     input_vids, 
                     target_prompt, 
@@ -66,8 +69,11 @@ def run_owl_batch(
                     fps_processed=fps_processed, 
                     scaling_factor=1/scaling_factor, 
                     batch_size=BATCH_SIZE, 
-                    save_dir=f"temp_{timestamp}", 
+                    save_dir=save_dir, 
                     progress=progress)
+    
+    # remove save_dir for storage 
+    shutil.rmtree(save_dir)
 
     end_time = time.time()
     print(f'Processing time: {end_time - start_time} seconds')

owl_batch.py

@@ -50,7 +50,7 @@ def owl_batch_video(
     df.to_csv(f"{save_dir}/detection_results.csv")
     
     # zip the save_dir
-    zip_file = f"{save_dir}/results.zip"
+    zip_file = f"results.zip"
     zip_directory(save_dir, zip_file)
 
     return zip_file