fix app

app.py

@@ -28,11 +28,7 @@ hf_hub_download(repo_id="VideoCrafter/Text2Video-1024", filename="model.ckpt", l
 # hf_hub_download(repo_id="MoonQiu/LongerCrafter", filename="model.pth", local_dir=ckpt_dir_256)
 
 
-def infer(prompt, output_size, seed):
-    num_frames = 32 
-    ddim_steps = 50
-    unconditional_guidance_scale = 12.0
-    save_fps = 10
+def infer(prompt, output_size, seed, num_frames, ddim_steps, unconditional_guidance_scale, save_fps):
     window_size = 16
     window_stride = 4
     
@@ -258,6 +254,28 @@ with gr.Blocks(css=css) as demo:
                 with gr.Row():
                     output_size = gr.Dropdown(["576x1024"], value="576x1024", label="Output Size (around 1800s for 576x1024)")
                     # output_size = gr.Dropdown(["576x1024", "256x256"], value="576x1024", label="Output Size", info="576x1024 is watermark-free")
+                with gr.Row():
+                    num_frames = gr.Slider(label='Frames',
+                             minimum=16,
+                             maximum=32,
+                             step=4,
+                             value=32)
+                    ddim_steps = gr.Slider(label='DDIM Steps',
+                             minimum=1,
+                             maximum=1000,
+                             step=1,
+                             value=50)
+                with gr.Row():
+                    unconditional_guidance_scale = gr.Slider(label='Unconditional Guidance Scale',
+                             minimum=0,
+                             maximum=50,
+                             step=1,
+                             value=12)
+                    save_fps = gr.Slider(label='Save FPS',
+                             minimum=1,
+                             maximum=30,
+                             step=1,
+                             value=10)
                 with gr.Row():
                     seed = gr.Slider(label='Random Seed',
                              minimum=0,
@@ -268,10 +286,10 @@ with gr.Blocks(css=css) as demo:
         submit_btn = gr.Button("Generate")
         video_result = gr.Video(label="Video Output")
 
-        gr.Examples(examples=examples, inputs=[prompt_in, output_size, seed])
+        gr.Examples(examples=examples, inputs=[prompt_in, output_size, seed, num_frames, ddim_steps, unconditional_guidance_scale, save_fps])
 
     submit_btn.click(fn=infer,
-            inputs=[prompt_in, output_size, seed],
+            inputs=[prompt_in, output_size, seed, num_frames, ddim_steps, unconditional_guidance_scale, save_fps],
             outputs=[video_result],
             api_name="zrscp")