Update app.py

app.py

@@ -38,8 +38,9 @@ def generate(input_video):
     temp_dir = tempfile.mkdtemp()
     
     workdir = temp_dir
-    pretrained_model = "pretrained/recon.safetensors"
-    num_frames = 1
+    recon_model = "pretrained/recon.safetensors"
+    interp_model = "pretrained/interp.safetensors"
+    num_frames = 16
     test_path = input_video
     
     try:
@@ -48,20 +49,19 @@ def generate(input_video):
             [
                 "python", "infer_3d.py", "big",
                 "--workspace", f"{workdir}",
-                "--resume", f"{pretrained_model}",
+                "--resume", f"{recon_model}",
+                "--interpresume", f"{interp_model}",
                 "--num_frames", f"{num_frames}",
                 "--test_path", f"{test_path}",
             ],
             check=True
         )
 
-        
-        # 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[0]
+   
     except subprocess.CalledProcessError as e:
-        return f"Error during inference: {str(e)}"
+        raise gr.Error(f"Error during inference: {str(e)}")
 
 with gr.Blocks() as demo:
     with gr.Column():