Update app.py

app.py

@@ -32,17 +32,17 @@ def init_predictor():
     )
     
 @spaces.GPU(duration=80)
-def generate_video(prompt, seed, image=None):
+def generate_video(prompt, seed, image=None, size, steps, frames):
     print(f"image:{type(image)}")
     if seed == -1:
         random.seed(time.time())
         seed = int(random.randrange(4294967294))
     kwargs = {
         "prompt": prompt,
-        "height": 320,
-        "width": 320,
-        "num_frames": 32,
-        "num_inference_steps": 20,
+        "height": size,
+        "width": size,
+        "num_frames": frames,
+        "num_inference_steps": steps,
         "seed": seed,
         "guidance_scale": 6.0,
         "embedded_guidance_scale": 1.0,
@@ -81,12 +81,18 @@ with gr.Blocks() as demo:
                     step=8,
                     value=64,
                 )
-                
+                steps =  = gr.Slider(
+                    label="Number of Steps",
+                    minimum=16,
+                    maximum=96,
+                    step=1,
+                    value=20,
+                )
             submit_button = gr.Button("Generate Video")
             output_video = gr.Video(label="Generated Video")
             submit_button.click(
                 fn=generate_video,
-                inputs=[prompt, seed, image],
+                inputs=[prompt, seed, image, size, steps, frames],
                 outputs=[output_video],
             )
 #        return demo