Update app.py

app.py

@@ -45,9 +45,12 @@ def process_video(video_file, mask_image, drawn_editor, progress=gr.Progress()):
     extracted_dir = os.path.join(base_dir, "extracted_frames")
     masked_dir = os.path.join(base_dir, "masked_frames")
     heatmap_dir = os.path.join(base_dir, "output_heatmap")
-    os.makedirs(extracted_dir, exist_ok=True)
-    os.makedirs(masked_dir, exist_ok=True)
-    os.makedirs(heatmap_dir, exist_ok=True)
+    
+    os.makedirs("video_outputs/extracted_frames", exist_ok=True)
+    os.makedirs("video_outputs/masked_frames", exist_ok=True)
+    os.makedirs("video_outputs/output_heatmap", exist_ok=True)
+    os.makedirs("video_inputs", exist_ok=True)
+    os.makedirs("assets", exist_ok=True)
 
     # Choose mask: from upload or drawing
     if mask_image is not None: