Update app_df.py

app_df.py

@@ -138,12 +138,16 @@ with gr.Blocks() as demo:
             with gr.Column():
 
                 prompt = gr.Textbox(label="Prompt")
-                model_id = gr.Dropdown(choices=model_options, value=model_options[0], label="Model ID")
-                resolution = gr.Radio(choices=resolution_options, value="540P", label="Resolution", interactive=False)
-                num_frames = gr.Slider(minimum=17, maximum=257, value=97, step=20, label="Number of Frames")
+
+                with gr.Row():
+                    target_length = gr.Radio(label="Video length target", choices=["4", "10", "15", "30", "60"], value="4")
+                
+                num_frames = gr.Slider(minimum=17, maximum=257, value=97, step=20, label="Number of Frames", interactive=False)
                 image = gr.Image(type="filepath", label="Input Image (optional)")
                 
                 with gr.Accordion("Advanced Settings", open=False):
+                    model_id = gr.Dropdown(choices=model_options, value=model_options[0], label="Model ID")
+                    resolution = gr.Radio(choices=resolution_options, value="540P", label="Resolution", interactive=False)
                     ar_step = gr.Number(label="AR Step", value=0)
                     causal_attention = gr.Checkbox(label="Causal Attention")
                     causal_block_size = gr.Number(label="Causal Block Size", value=1)
@@ -178,6 +182,31 @@ with gr.Blocks() as demo:
                     inputs = [prompt, image]
                 )
 
+    def set_num_frames(target_l):
+
+        n_frames = 0
+
+        if target_l == "4":
+            n_frames = 97
+        elif target_l == "10":
+            n_frames = 257
+        elif target_l == "15":
+            n_frames = 377
+        elif target_l == "30":
+            n_frames = 737
+        elif target_l == "60":
+            n_frames = 1457
+        
+        return n_frames
+        
+
+    target_length.change(
+        fn = set_num_frames,
+        inputs = [target_length],
+        outputs = [num_frames],
+        queue = False
+    )
+
     submit_btn.click(
         fn = generate_diffusion_forced_video,
         inputs = [