Update app.py

app.py

@@ -25,6 +25,7 @@ pipe.to("cuda")
 causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
 pipe.load_lora_weights(causvid_path, adapter_name="causvid_lora")
 pipe.set_adapters(["causvid_lora"], adapter_weights=[0.95])
+pipe.fuse_lora()
 
 MOD_VALUE = 32
 DEFAULT_H_SLIDER_VALUE = 512
@@ -78,7 +79,19 @@ def handle_image_upload_for_dims_wan(uploaded_pil_image, current_h_val, current_
         gr.Warning("Error attempting to calculate new dimensions")
         return gr.update(value=DEFAULT_H_SLIDER_VALUE), gr.update(value=DEFAULT_W_SLIDER_VALUE)
 
-@spaces.GPU
+def get_duration(input_image, prompt, height, width, 
+                   negative_prompt=default_negative_prompt, duration_seconds = 2,
+                   guidance_scale = 1, steps = 4,
+                   seed = 42, randomize_seed = False, 
+                   progress=gr.Progress(track_tqdm=True)):
+    if steps > 4 and duration_seconds > 2:
+        return 90
+    elif steps > 4 or duration_seconds > 2:
+        return 75
+    else:
+        return 60
+
+@spaces.GPU(duration=get_duration)
 def generate_video(input_image, prompt, height, width, 
                    negative_prompt=default_negative_prompt, duration_seconds = 2,
                    guidance_scale = 1, steps = 4,