Update app.py

app.py

@@ -23,7 +23,7 @@ from optimization import optimize_pipeline_
 MODEL_ID = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
 
 # --- NEW: Flexible Dimension Constants ---
-MAX_DIMENSION = 832
+MAX_DIMENSION = 720
 MIN_DIMENSION = 480
 DIMENSION_MULTIPLE = 16
 SQUARE_SIZE = 480
@@ -137,14 +137,14 @@ def resize_and_crop_to_match(target_image, reference_image):
     left, top = (new_width - ref_width) // 2, (new_height - ref_height) // 2
     return resized.crop((left, top, left + ref_width, top + ref_height))
 
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=40)
 def generate_video(
     start_image_pil,
     end_image_pil,
     prompt,
     negative_prompt=default_negative_prompt,
-    duration_seconds=2.1,
-    steps=8,
+    duration_seconds=2.5,
+    steps=5,
     guidance_scale=1,
     guidance_scale_2=1,
     seed=42,
@@ -219,9 +219,9 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as app:
                 prompt = gr.Textbox(label="Prompt", info="Describe the transition between the two images")
 
                 with gr.Accordion("Advanced Settings", open=False):
-                    duration_seconds_input = gr.Slider(minimum=MIN_DURATION, maximum=MAX_DURATION, step=0.1, value=2.1, label="Video Duration (seconds)", info=f"Clamped to model's {MIN_FRAMES_MODEL}-{MAX_FRAMES_MODEL} frames at {FIXED_FPS}fps.")
+                    duration_seconds_input = gr.Slider(minimum=MIN_DURATION, maximum=MAX_DURATION, step=0.1, value=2.5, label="Video Duration (seconds)", info=f"Clamped to model's {MIN_FRAMES_MODEL}-{MAX_FRAMES_MODEL} frames at {FIXED_FPS}fps.")
                     negative_prompt_input = gr.Textbox(label="Negative Prompt", value=default_negative_prompt, lines=3)
-                    steps_slider = gr.Slider(minimum=1, maximum=30, step=1, value=4, label="Inference Steps")
+                    steps_slider = gr.Slider(minimum=1, maximum=30, step=1, value=5, label="Inference Steps")
                     guidance_scale_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=1.0, label="Guidance Scale - high noise")
                     guidance_scale_2_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=1.0, label="Guidance Scale - low noise")
                     with gr.Row():
@@ -255,17 +255,6 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as app:
         outputs=ui_outputs
     )
 
-    gr.Examples(
-        examples=[
-            ["poli_tower.png", "tower_takes_off.png", "the man turns around"],
-            ["ugly_sonic.jpeg", "squatting_sonic.png", "the character dodges the missiles"],
-            ["capyabara_zoomed.png", "capyabara.webp", "a dramatic dolly zoom"],
-        ],
-        inputs=[start_image, end_image, prompt],
-        outputs=ui_outputs,
-        fn=generate_video,
-        cache_examples="lazy",
-    )
 
 if __name__ == "__main__":
     app.launch(share=True)