Update app.py

app.py

@@ -334,7 +334,7 @@ logger.info("Manual diff_b/diff patches applied.")
 
 # --- Gradio Interface Function ---
 @spaces.GPU
-def generate_video(prompt, negative_prompt, height, width, num_frames, guidance_scale, fps):
+def generate_video(prompt, negative_prompt, height, width, num_frames, guidance_scale, steps, fps):
     logger.info("Starting video generation...")
     logger.info(f"  Prompt: {prompt}")
     logger.info(f"  Negative Prompt: {negative_prompt if negative_prompt else 'None'}")
@@ -355,6 +355,7 @@ def generate_video(prompt, negative_prompt, height, width, num_frames, guidance_
             width=width,
             num_frames=num_frames,
             guidance_scale=float(guidance_scale),
+            num_inference_steps=steps
         ).frames
 
     if not output_frames_list or not output_frames_list[0]:
@@ -393,6 +394,7 @@ with gr.Blocks() as demo:
             with gr.Row():
                 num_frames_input = gr.Slider(minimum=16, maximum=100, step=1, value=25, label="Number of Frames") # Reduced default for faster demo
                 fps_input = gr.Slider(minimum=5, maximum=30, step=1, value=15, label="Output FPS")
+            steps = gr.Slider(minimum=1.0, maximum=30.0, value=4.0, label="Steps")
             guidance_scale_input = gr.Slider(minimum=1.0, maximum=20.0, step=0.5, value=5.0, label="Guidance Scale")
 
             generate_button = gr.Button("Generate Video", variant="primary")
@@ -416,12 +418,12 @@ with gr.Blocks() as demo:
 
     gr.Examples(
         examples=[
-            ["A panda eating bamboo in a lush forest, cinematic lighting", default_negative_prompt, 480, 832, 25, 5.0, 15],
-            ["A majestic eagle soaring over snowy mountains", default_negative_prompt, 512, 768, 30, 7.0, 12],
-            ["Timelapse of a flower blooming, vibrant colors", "static, ugly", 384, 640, 40, 6.0, 20],
-            ["Astronaut walking on the moon, Earth in the background, highly detailed", default_negative_prompt, 480, 832, 20, 5.5, 10],
+            ["A panda eating bamboo in a lush forest, cinematic lighting", default_negative_prompt, 480, 832, 25, 5.0, 4, 15],
+            ["A majestic eagle soaring over snowy mountains", default_negative_prompt, 512, 768, 30, 7.0, 4, 12],
+            ["Timelapse of a flower blooming, vibrant colors", "static, ugly", 384, 640, 40, 6.0, 4, 20],
+            ["Astronaut walking on the moon, Earth in the background, highly detailed", default_negative_prompt, 480, 832, 20, 5.5, 4, 10],
         ],
-        inputs=[prompt_input, negative_prompt_input, height_input, width_input, num_frames_input, guidance_scale_input, fps_input],
+        inputs=[prompt_input, negative_prompt_input, height_input, width_input, num_frames_input, guidance_scale_input, steps, fps_input],
         outputs=video_output,
         fn=generate_video,
         cache_examples=False