Update app.py

app.py

@@ -35,8 +35,9 @@ video_examples = [
 
 def generate(input_video):
 
-    #--test_path data_test/otter-on-surfboard_fg.mp4
-    workdir = "results"
+    temp_dir = tempfile.mkdtemp()
+    
+    workdir = temp_dir
     pretrained_model = "pretrained/recon.safetensors"
     num_frames = 1
     test_path = input_video
@@ -58,7 +59,7 @@ def generate(input_video):
         # Retrieve the file name without the extension
         #removed_bg_file_name = os.path.splitext(os.path.basename(removed_bg_path))[0]
         output_videos = glob(os.path.join(f"{workdir}", "*.mp4"))
-        return output_videos
+        return output_videos[0]
     except subprocess.CalledProcessError as e:
         return f"Error during inference: {str(e)}"