Update gradio_app.py

gradio_app.py

@@ -31,6 +31,17 @@ snapshot_download(repo_id="stabilityai/sd-vae-ft-mse", local_dir="./weights/sd-v
 from diffueraser.diffueraser import DiffuEraser
 from propainter.inference import Propainter, get_device
 
+base_model_path = "weights/stable-diffusion-v1-5"
+vae_path = "weights/sd-vae-ft-mse"
+diffueraser_path = "weights/diffuEraser"
+propainter_model_dir = "weights/propainter"
+
+# Model setup
+device = get_device()
+ckpt = "2-Step"
+video_inpainting_sd = DiffuEraser(device, base_model_path, vae_path, diffueraser_path, ckpt=ckpt)
+propainter = Propainter(propainter_model_dir, device=device)
+
 # Helper function to trim videos
 def trim_video(input_path, output_path, max_duration=5):
     clip = VideoFileClip(input_path)
@@ -39,7 +50,7 @@ def trim_video(input_path, output_path, max_duration=5):
     clip.close()
     trimmed_clip.close()
 
-@spaces.GPU(duration=180)
+@spaces.GPU(duration=120)
 def infer(input_video, input_mask):
     # Setup paths and parameters
     save_path = "results"
@@ -47,12 +58,7 @@ def infer(input_video, input_mask):
     max_img_size = 960
     ref_stride = 10
     neighbor_length = 10
-    subvideo_length = 50
-
-    base_model_path = "weights/stable-diffusion-v1-5"
-    vae_path = "weights/sd-vae-ft-mse"
-    diffueraser_path = "weights/diffuEraser"
-    propainter_model_dir = "weights/propainter"
+    subvideo_length = 50 
 
     if not os.path.exists(save_path):
         os.makedirs(save_path)
@@ -74,12 +80,6 @@ def infer(input_video, input_mask):
     clip.close()
     video_length = int(video_duration * 30)
 
-    # Model setup
-    device = get_device()
-    ckpt = "2-Step"
-    video_inpainting_sd = DiffuEraser(device, base_model_path, vae_path, diffueraser_path, ckpt=ckpt)
-    propainter = Propainter(propainter_model_dir, device=device)
-
     # Run models
     start_time = time.time()