Callable ZeroGPU duration; move theme/css to launch() for Gradio 6

app.py

@@ -242,7 +242,21 @@ def _parse_labels(labels_text: str):
     return names
 
 
-@spaces.GPU(duration=60)
+def _estimate_duration(shape_file=None, labels_text="", num_points=pa.DEFAULT_NPOINTS, *args, **kwargs):
+    """GPU reservation. Measured worst case is ~1.2 s of compute at the heaviest settings; the rest is
+    headroom for loading / surface-sampling a large user-supplied mesh inside the same call."""
+    try:
+        mb = os.path.getsize(shape_file) / 1e6
+    except Exception:
+        mb = 0.0
+    try:
+        n = int(num_points)
+    except Exception:
+        n = pa.DEFAULT_NPOINTS
+    return int(min(90, 8 + 0.6 * mb + 2.0 * n / pa.DEFAULT_NPOINTS))
+
+
+@spaces.GPU(duration=_estimate_duration)
 def segment(
     shape_file: str,
     labels_text: str = "back, seat, leg, arm",
@@ -331,7 +345,7 @@ EXAMPLES = [
     ["examples/bunny.obj", "ear, head, torso, foot"],
 ]
 
-with gr.Blocks(theme=gr.themes.Citrus(), css=CSS, title="PatchAlign3D") as demo:
+with gr.Blocks(title="PatchAlign3D") as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
             """
@@ -428,4 +442,4 @@ work best; the Bunny is there to show that arbitrary meshes go through the same
     labels_text.submit(segment, inputs=inputs, outputs=outputs)
 
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    demo.launch(theme=gr.themes.Citrus(), css=CSS, mcp_server=True)