fix app

app.py

@@ -28,8 +28,7 @@ os.makedirs(ckpt_dir_256, exist_ok=True)
 hf_hub_download(repo_id="MoonQiu/LongerCrafter", filename="model.pth", local_dir=ckpt_dir_256)
 
 
-def infer(prompt):
-    output_size = "256x256"
+def infer(prompt, output_size):
     num_frames = 32 
     ddim_steps = 50
     unconditional_guidance_scale = 12.0
@@ -159,13 +158,12 @@ def infer(prompt):
 examples = [
     "A chihuahua in astronaut suit floating in space, cinematic lighting, glow effect",
     "Campfire at night in a snowy forest with starry sky in the background",
-    "A dark knight riding a black horse on the grassland, in sunset",
     "A corgi is swimming quickly",
     "A panda is surfing in the universe",
 ]
 
 css = """
-#col-container {max-width: 510px; margin-left: auto; margin-right: auto;}
+#col-container {max-width: 640px; margin-left: auto; margin-right: auto;}
 a {text-decoration-line: underline; font-weight: 600;}
 .animate-spin {
   animation: spin 1s linear infinite;
@@ -259,15 +257,14 @@ with gr.Blocks(css=css) as demo:
         )
 
         prompt_in = gr.Textbox(label="Prompt", placeholder="A chihuahua in astronaut suit floating in space, cinematic lighting, glow effect", elem_id="prompt-in")
-        #neg_prompt = gr.Textbox(label="Negative prompt", value="text, watermark, copyright, blurry, nsfw", elem_id="neg-prompt-in")
-        #inference_steps = gr.Slider(label="Inference Steps", minimum=10, maximum=100, step=1, value=40, interactive=False)
+        output_size = gr.Radio(["576x1024", "256x256"], label="Output Size", info="576x1024 is watermark-free"),
         submit_btn = gr.Button("Generate")
         video_result = gr.Video(label="Video Output", elem_id="video-output")
 
         gr.Examples(examples=examples, inputs=[prompt_in])
 
     submit_btn.click(fn=infer,
-            inputs=[prompt_in],
+            inputs=[prompt_in, output_size],
             outputs=[video_result],
             api_name="zrscp")