Update app.py

app.py

@@ -515,7 +515,33 @@ def find_diagnostic(output_dir: Path, output_path: Path, kind: str) -> Optional[
 STARTUP_OK, STARTUP_LOG = startup_preflight()
 
 
-@spaces.GPU(duration=3600, size="xlarge")
+def estimate_gpu_duration(
+    preset_name: str,
+    source_mode: str,
+    sample_name: Optional[str],
+    dataset_zip: Optional[Any],
+    env_mode: str,
+    env_name: Optional[str],
+    env_zip: Optional[Any],
+    steps: int,
+    cfg_scale: float,
+    quality: int,
+    wo_ref_weight: float,
+    drop_mr: bool,
+    use_multi_ref: bool,
+) -> int:
+    preset = PRESETS.get(preset_name, PRESETS["Basic 25-frame relighting"])
+    frames = int(preset["frames"])
+    steps = int(steps)
+
+    if frames == 1:
+        return min(150, max(45, 25 + steps * 2))
+    if frames <= 25:
+        return min(900, max(180, 90 + steps * 10))
+    return min(1500, max(300, 180 + steps * 18))
+
+
+@spaces.GPU(duration=estimate_gpu_duration, size="large")
 def run_inference(
     preset_name: str,
     source_mode: str,
@@ -985,4 +1011,4 @@ if __name__ == "__main__":
         server_name=args.server_name,
         server_port=args.server_port,
         show_error=True,
-    )
+    )