Update app.py

app.py

@@ -49,6 +49,8 @@ def generate(
     seed: Optional[int] = -1,
     motion_bucket_id: int = 127,
     fps_id: int = 25,
+    num_inference_steps: int = 25,
+    num_frames: int = 128,
     output_folder: str = "outputs",
     progress=gr.Progress(track_tqdm=True)):
     
@@ -65,12 +67,12 @@ def generate(
 
     frames = pipe(
         input_image=image.resize((512, 512)), 
-        num_frames=128, 
+        num_frames=num_frames, 
         fps=fps_id, 
         height=512, 
         width=512,
         motion_bucket_id=motion_bucket_id,
-        num_inference_steps=50,
+        num_inference_steps=num_inference_steps,
         min_cfg_scale=2, 
         max_cfg_scale=2, 
         contrast_enhance_scale=1.2
@@ -110,20 +112,36 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
                     label="Motion bucket id", 
                     info="Controls how much motion to add/remove from the image", 
                     value=127, 
+                    step=1,
                     minimum=1, 
                     maximum=255
                 )
                 fps_id = gr.Slider(
                     label="Frames per second", 
                     info="The length of your video in seconds will be 25/fps", 
-                    value=6, 
+                    value=6,
+                    step=1,
                     minimum=5, 
                     maximum=30
                 )
+                num_inference_steps = gr.Slider(
+                    label="Inference steps", 
+                    info="Inference steps", 
+                    value=25, 
+                    minimum=1, 
+                    maximum=50
+                )
+                num_frames = gr.Slider(
+                    label="Frames num", 
+                    info="Frames num", 
+                    value=25, 
+                    minimum=1, 
+                    maximum=50
+                )               
     with gr.Row():
         submit_btn = gr.Button(value="Generate")
-        stop_btn = gr.Button(variant="stop")
-        clear_btn = gr.ClearButton([image, seed, video])
+        stop_btn = gr.Button(value="Stop", variant="stop")
+        #clear_btn = gr.ClearButton([image, seed, video])
     gr.Examples(
         examples=examples,
         inputs=image,
@@ -133,7 +151,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
         examples_per_page=4,
     )
 
-    submit_event = submit_btn.click(fn=generate, inputs=[image, seed, motion_bucket_id, fps_id], outputs=[video, seed], api_name="video")
-    stop_btn.click(lambda _: gr.update(interactive=False), None, stop_btn, cancels=[submit_event])
+    submit_event = submit_btn.click(fn=generate, inputs=[image, seed, motion_bucket_id, fps_id,num_inference_steps, num_frames], outputs=[video, seed], api_name="video")
+    #stop_btn.click(lambda _: gr.update(interactive=False), None, stop_btn, cancels=[submit_event])
     
 demo.queue().launch()