fix: duration and input example

app.py

@@ -74,7 +74,15 @@ def get_examples():
     ]
     return case
 
-@spaces.GPU(duration=240)
+def get_duration(*args, **kwargs) -> float:
+    model_type_val = kwargs.get("model_type_val", 'FLUX')
+    n_max_val = kwargs.get("n_max_val", 15)
+    flowopt_iterations_val = kwargs.get("flowopt_iterations_val", 8)
+
+    step_duration = 1 if model_type_val == 'FLUX' else 0.4
+    return (n_max_val * (flowopt_iterations_val + 2) * step_duration) + 40
+
+@spaces.GPU(duration=get_duration)
 def FlowOpt_run(
     image_src_val: str, model_type_val: str, T_steps_val: int,
     n_max_val: int, eta_val: float, flowopt_iterations_val: int,
@@ -159,9 +167,9 @@ with gr.Blocks(css=css) as demo:
 
         with gr.Row():
             with gr.Column():
-                image_src = gr.Image(type="filepath", label="Source Image", value="inputs/cat.png",)
-                src_prompt = gr.Textbox(lines=2, label="Source Prompt", value="A cat sitting in the grass")
-                tar_prompt = gr.Textbox(lines=2, label="Target Prompt", value="A puppy sitting in the grass")
+                image_src = gr.Image(type="filepath", label="Source Image", value="inputs/corgi_walking.png",)
+                src_prompt = gr.Textbox(lines=2, label="Source Prompt", value="A cute brown and white dog walking on a sidewalk near a body of water. The dog is wearing a pink vest, adding a touch of color to the scene.")
+                tar_prompt = gr.Textbox(lines=2, label="Target Prompt", value="A cute brown and white dog walking on a sidewalk near a body of water. The dog is wearing a pink vest, adding a touch of color to the scene. The dog and sidewalk are constructed entirely out of Lego bricks, showcasing a blocky and geometric appearance.")
                 submit_button = gr.Button("Run FlowOpt", variant="primary")
 
                 with gr.Row():
@@ -169,7 +177,7 @@ with gr.Blocks(css=css) as demo:
                     T_steps = gr.Slider(value=15, minimum=10, maximum=50, step=1, label="Total Steps", info="Total number of discretization steps.")
                     n_max = gr.Slider(value=13, minimum=1, maximum=15, step=1, label="n_max", info="Control the strength of the edit.")
                     eta = gr.Slider(value=0.0025, minimum=0.0001, maximum=0.05, label="eta", info="Control the optimization step-size.")
-                    flowopt_iterations = gr.Number(value=10, minimum=1, maximum=15, label="flowopt_iterations", info="Max number of FlowOpt iterations")
+                    flowopt_iterations = gr.Number(value=8, minimum=1, maximum=15, label="flowopt_iterations", info="Max number of FlowOpt iterations")
 
             with gr.Column():
                 image_tar = gr.Gallery(