Update app_exp.py

app_exp.py

@@ -129,6 +129,25 @@ def torch_gc():
         torch.cuda.empty_cache()
         torch.cuda.ipc_collect()
 
+def check_duration(
+    mode,
+    prompt, 
+    neg_prompt, 
+    image,
+    height, width, resolution,
+    seed,
+    use_distill,
+    use_refine,
+    progress
+):
+    if use_distill and resolution=="480p":
+        return 180
+    elif resolution=="720p":
+        return 360
+    else:
+        return 900
+
+@spaces.GPU(duration=180)
 def generate_video(mode, prompt, neg_prompt, image, height, width, resolution,
                    seed, use_distill, use_refine, duration_sec, progress=gr.Progress(track_tqdm=True)):