Update app.py

app.py

@@ -12,18 +12,19 @@ pipe.enable_model_cpu_offload()
 pipe.enable_vae_slicing()
 
 @spaces.GPU(duration=250)
-def generate(prompt, num_inference_steps=25, frames=200):
+def generate(prompt, num_inference_steps, num_frames):
     video_frames = pipe(prompt, num_inference_steps, num_frames).frames
     video_path = export_to_video(video_frames)
     return video_path
 
-prompt = gr.Textbox("Enter prompt to generate a video")
+prompt = gr.Textbox(label="Enter prompt to generate a video")
 num_inference_steps = gr.Slider(10, 50, value=25)
+num_frames = gr.Slider(100, 500, value=200)
 
 interface = gr.Interface(
   generate,
-  inputs=[prompt, num_inference_steps],
-  examples=[["Astronaut riding a horse", 25], ["Darth vader surfing in waves", 20]],
+  inputs=[prompt, num_inference_steps, num_frames],
+  examples=[["Astronaut riding a horse", 25, 200], ["Darth vader surfing in waves", 20, 100]],
   outputs="video",
   cache_examples=False,
   theme="soft"