Update app.py

app.py

@@ -13,9 +13,14 @@ def infer(video_frames, masks_frames):
     masks_folder = "inputs/object_removal/bmx-trees_mask"
 
     # Create the "results" folder if it doesn't exist
-    output_folder = "results/bmx-trees"
+    output_folder = "results"
     if not os.path.exists(output_folder):
         os.makedirs(output_folder)
+
+    # Ensure the "bmx-trees" directory is also created if it doesn't exist
+    bmx_trees_folder = os.path.join(output_folder, "bmx-trees")
+    if not os.path.exists(bmx_trees_folder):
+        os.makedirs(bmx_trees_folder)
     
     command = [
       f"python", 
@@ -28,7 +33,7 @@ def infer(video_frames, masks_frames):
     execute_command(command)
 
     # Get the list of files in the "results" folder
-    result_files = os.listdir(output_folder)
+    result_files = os.listdir(bmx_trees_folder)
 
     return result_files