Update app.py

app.py

@@ -66,8 +66,8 @@ model = load_model()
 # Text-to-video generation function
 @spaces.GPU(duration=120)
 def generate_video(prompt, image=None, duration=5, guidance_scale=9, video_guidance_scale=5, progress=gr.Progress(track_tqdm=True)):
-    multiplier = 0.8 if is_canonical else 2.4
-    temp = int(duration * 0.8)  # Convert seconds to temp value (assuming 24 FPS)
+    multiplier = 1.2 if is_canonical else 3.0
+    temp = int(duration * multiplier) + 1 
     torch_dtype = torch.bfloat16 if MODEL_DTYPE == "bf16" else torch.float32
     if(image):
         cropped_image = center_crop(image, 1280, 720)
@@ -98,7 +98,7 @@ def generate_video(prompt, image=None, duration=5, guidance_scale=9, video_guida
                 save_memory=True,
             )
     output_path = f"{str(uuid.uuid4())}_output_video.mp4"
-    export_to_video(frames, output_path, fps=8 if is_canonical else 24)
+    export_to_video(frames, output_path, fps=24)
     return output_path
 
 # Gradio interface
@@ -113,7 +113,7 @@ with gr.Blocks() as demo:
                 i2v_image = gr.Image(type="pil", label="Input Image")
             t2v_prompt = gr.Textbox(label="Prompt")
             with gr.Accordion("Advanced settings", open=False):
-                t2v_duration = gr.Slider(minimum=1, maximum=10, value=5, step=1, label="Duration (seconds)", visible=not is_canonical)
+                t2v_duration = gr.Slider(minimum=1, maximum=10, value=2 if is_canonical else 5, step=1, label="Duration (seconds)", visible=not is_canonical)
                 t2v_guidance_scale = gr.Slider(minimum=1, maximum=15, value=9, step=0.1, label="Guidance Scale")
                 t2v_video_guidance_scale = gr.Slider(minimum=1, maximum=15, value=5, step=0.1, label="Video Guidance Scale")
             t2v_generate_btn = gr.Button("Generate Video")