Update app.py

app.py

@@ -29,11 +29,11 @@ pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 max_64_bit_int = 2**63 - 1
 
 def generate_video(
+    secret_token: str,
     image: Image,
     seed: int,
     motion_bucket_id: int = 127,
     fps_id: int = 6,
-    secret_token: str = "",
     version: str = "svd_xt",
     cond_aug: float = 0.02,
     decoding_t: int = 3,  # Number of frames decoded at a time! This eats most VRAM. Reduce if necessary.
@@ -103,35 +103,31 @@ def resize_image(image, output_size=(1024, 576)):
     return cropped_image
 
 with gr.Blocks() as demo:
-  secret_token = gr.Text(
-    label='Secret Token',
-    max_lines=1,
-    placeholder='Enter your secret token',
-  )
-  gr.HTML("""
-    <div style="z-index: 100; position: fixed; top: 0px; right: 0px; left: 0px; bottom: 0px; width: 100%; height: 100%; background: white; display: flex; align-items: center; justify-content: center; color: black;">
-      <div style="text-align: center; color: black;">
-        <p style="color: black;">This space is a REST API to programmatically generate MP4 videos.</p>
-        <p style="color: black;">Interested in using it? Look no further than the <a href="https://huggingface.co/spaces/multimodalart/stable-video-diffusion" target="_blank">original space</a>!</p>
-      </div>
-    </div>""")
-  with gr.Row():
-    with gr.Column():
-        image = gr.Image(label="Upload your image", type="pil")
-        generate_btn = gr.Button("Generate")
-        base64_out = gr.Textbox(label="Base64 Video")
-
-  with gr.Accordion("Advanced options", open=False):
-      seed = gr.Slider(label="Seed", value=42, randomize=False, minimum=0, maximum=max_64_bit_int, step=1)
-      motion_bucket_id = gr.Slider(label="Motion bucket id", info="Controls how much motion to add/remove from the image", value=127, 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, minimum=5, maximum=30)
+    secret_token = gr.Text(
+        label='Secret Token',
+        max_lines=1,
+        placeholder='Enter your secret token',
+   )
+   gr.HTML("""
+       <div style="z-index: 100; position: fixed; top: 0px; right: 0px; left: 0px; bottom: 0px; width: 100%; height: 100%; background: white; display: flex; align-items: center; justify-content: center; color: black;">
+       <div style="text-align: center; color: black;">
+       <p style="color: black;">This space is a REST API to programmatically generate MP4 videos.</p>
+       <p style="color: black;">Interested in using it? Look no further than the <a href="https://huggingface.co/spaces/multimodalart/stable-video-diffusion" target="_blank">original space</a>!</p>
+       </div>
+       </div>"""
+   )
+   image = gr.Image(label="Upload your image", type="pil")
+   generate_btn = gr.Button("Generate")
+   base64_out = gr.Textbox(label="Base64 Video")
+   seed = gr.Slider(label="Seed", value=42, randomize=False, minimum=0, maximum=max_64_bit_int, step=1)
+   motion_bucket_id = gr.Slider(label="Motion bucket id", info="Controls how much motion to add/remove from the image", value=127, 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, minimum=5, maximum=30)
       
-  generate_btn.click(
-      fn=generate_video,
-      inputs=[image, seed, motion_bucket_id, fps_id, secret_token],
-      outputs=base64_out,
-      api_name="run"
-  )
-
-if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+    generate_btn.click(
+        fn=generate_video,
+        inputs=[secret_token, image, seed, motion_bucket_id, fps_id],
+        outputs=base64_out,
+        api_name="run"
+    )
+
+demo.queue(max_size=20).launch()