Update app.py

app.py

@@ -117,7 +117,23 @@ def torch_gc():
         torch.cuda.empty_cache()
         torch.cuda.ipc_collect()
 
-@spaces.GPU(duration=900)
+def check_duration(
+    mode,
+    prompt, 
+    neg_prompt, 
+    image,
+    height, width, resolution,
+    seed,
+    use_distill,
+    use_refine,
+):
+    if use_refine:
+        return 200
+    else:
+        return 900
+    
+
+@spaces.GPU(check_duration)
 def generate_video(
     mode,
     prompt, 
@@ -255,7 +271,7 @@ with gr.Blocks(css="style.css") as demo:
                         width_t2v = gr.Slider(label="Width", minimum=256, maximum=1024, value=832, step=64)
                     with gr.Row():
                         seed_t2v = gr.Number(label="Seed", value=42, precision=0)
-                        distill_t2v = gr.Checkbox(label="Use Distill Mode", value=False, info="Faster, lower quality base generation.")
+                        distill_t2v = gr.Checkbox(label="Use Distill Mode", value=True, info="Faster, lower quality base generation.")
                         refine_t2v = gr.Checkbox(label="Use Refine Mode", value=False, info="Higher quality & resolution, but slower. Uses Distill mode for its first stage.")
                     
                     t2v_button = gr.Button("Generate Video", variant="primary")