add 512

app.py

@@ -55,6 +55,7 @@ def infer(prompt, output_size, seed, num_frames, ddim_steps, unconditional_guida
         model_1024.eval()
         model = model_1024
         fps = 28
+        num_frames = min(num_frames, 36)
     # elif output_size == "256x256":
     #     width = 256 
     #     height = 256
@@ -258,9 +259,11 @@ with gr.Blocks(css=css) as demo:
             """
             <h1 style="text-align: center;">LongerCrafter(FreeNoise) Text-to-Video</h1>
             <p style="text-align: center;">
-            Tuning-Free Longer Video Diffusion via Noise Rescheduling <br />
+            Tuning-Free Longer Video Diffusion via Noise Rescheduling
             </p>
-                        
+            <p style="text-align: center;">
+            [Arxiv](https://arxiv.org/abs/2310.15169) | [Project Page](http://haonanqiu.com/projects/FreeNoise.html) | [Github](https://github.com/arthur-qiu/LongerCrafter) <br />
+            </p>            
             """
         )
 
@@ -271,9 +274,9 @@ with gr.Blocks(css=css) as demo:
                 with gr.Row():
                     output_size = gr.Dropdown(["320x512", "576x1024"], value="320x512", label="Output Size", info="around 350s for 320x512, around 900s for 576x1024")
                 with gr.Row():
-                    num_frames = gr.Slider(label='Frames (a multiple of 4)',
+                    num_frames = gr.Slider(label='Frames (a multiple of 4), max 36 for 1024 model',
                              minimum=16,
-                             maximum=36,
+                             maximum=64,
                              step=4,
                              value=32)
                     ddim_steps = gr.Slider(label='DDIM Steps',
@@ -299,7 +302,7 @@ with gr.Blocks(css=css) as demo:
                              step=1,
                              value=123)
 
-        submit_btn = gr.Button("Generate")
+        submit_btn = gr.Button("Generate", variant='primary')
         video_result = gr.Video(label="Video Output")
 
         gr.Examples(examples=examples, inputs=[prompt_in, output_size, seed, num_frames, ddim_steps, unconditional_guidance_scale, save_fps])