fix: duration

app.py

@@ -74,13 +74,14 @@ def get_examples():
     ]
     return case
 
-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
+def get_duration(
+    image_src_val: str, model_type_val: str, T_steps_val: int,
+    n_max_val: int, eta_val: float, flowopt_iterations_val: int,
+    src_prompt_val: str, tar_prompt_val: str,
+    src_guidance_scale_val: float, tar_guidance_scale_val: float,
+) -> float:
+    step_duration = 0.6 if model_type_val == 'FLUX' else 0.2
+    return (n_max_val * (flowopt_iterations_val + 2) * step_duration) + 20
 
 @spaces.GPU(duration=get_duration)
 def FlowOpt_run(