Right-size ZeroGPU duration from mesh header, fix Gradio 6 theme/css, add second example scan

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 examples/attic.glb filter=lfs diff=lfs merge=lfs -text
+examples/historic-interior.glb filter=lfs diff=lfs merge=lfs -text

app.py

@@ -12,6 +12,7 @@ os.environ.setdefault("NUMBA_CACHE_DIR", "/tmp/numba-cache")
 
 import spaces                                             # noqa: E402  (before torch)
 
+import json                                               # noqa: E402
 import tempfile                                           # noqa: E402
 import time                                               # noqa: E402
 from typing import List, Optional, Tuple                  # noqa: E402
@@ -46,7 +47,54 @@ def _empty(msg: str):
     return None, [], msg
 
 
-@spaces.GPU(duration=110)
+# --------------------------------------------------------------------------- #
+# GPU-duration estimate
+#
+# Runtime is dominated by the vertex count (superpoint graph segmentation, the
+# 0.6 x |superpoints| adaptive queries and their NMS).  Measured on this Space:
+# 211k vertices -> 8.2 s, 526k vertices -> 19.2 s, i.e. ~0.035 s per 1k vertices.
+# The vertex count is read straight out of the file header (cheap, no parsing of
+# the geometry) so every visitor only reserves the quota their own scan needs.
+# --------------------------------------------------------------------------- #
+def _vertex_count(path: str) -> Optional[int]:
+    """Vertex count from a glTF-binary / PLY header, without loading geometry."""
+    try:
+        with open(path, "rb") as fh:
+            head = fh.read(20)
+            if head[:4] == b"glTF":
+                chunk_len = int.from_bytes(head[12:16], "little")
+                if head[16:20] != b"JSON":
+                    return None
+                gltf = json.loads(fh.read(chunk_len).decode("utf-8", "replace"))
+                accessors = gltf.get("accessors", [])
+                total = 0
+                for mesh in gltf.get("meshes", []):
+                    for prim in mesh.get("primitives", []):
+                        i = prim.get("attributes", {}).get("POSITION")
+                        if isinstance(i, int) and 0 <= i < len(accessors):
+                            total += int(accessors[i].get("count", 0))
+                return total or None
+            if head[:3] == b"ply":
+                fh.seek(0)
+                for line in fh.read(8192).split(b"\n"):
+                    if line.startswith(b"element vertex"):
+                        return int(line.split()[2])
+    except Exception:
+        pass
+    return None
+
+
+def _gpu_duration(mesh_file: Optional[str], *args, **kwargs) -> int:
+    if not mesh_file or not os.path.exists(mesh_file):
+        return 25
+    verts = _vertex_count(mesh_file)
+    if verts is None:                       # OBJ / unknown container: size proxy
+        verts = os.path.getsize(mesh_file) / 60.0
+    seconds = (1.0 + 0.035 * verts / 1000.0) * 1.5 + 5.0   # fit + 50% + fork cost
+    return int(min(75, max(20, round(seconds))))
+
+
+@spaces.GPU(duration=_gpu_duration)
 def segment(
     mesh_file: Optional[str],
     up_axis: str = "Auto",
@@ -116,7 +164,7 @@ CSS = """
 #viewer { height: 520px; }
 """
 
-with gr.Blocks(theme=gr.themes.Citrus(), css=CSS, title="AQ3D 3D Instance Segmentation") as demo:
+with gr.Blocks(title="AQ3D 3D Instance Segmentation") as demo:
     gr.Markdown(
         """
         # 🪑 AQ3D — 3D Instance Segmentation
@@ -161,6 +209,7 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS, title="AQ3D 3D Instance Segmen
     gr.Examples(
         examples=[
             ["examples/attic.glb"],
+            ["examples/historic-interior.glb"],
         ],
         inputs=[mesh_in],
         outputs=[mesh_out, table, status],
@@ -193,4 +242,5 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS, title="AQ3D 3D Instance Segmen
     )
 
 if __name__ == "__main__":
-    demo.queue().launch(mcp_server=True)
+    # Gradio 6 moved `theme` / `css` from the Blocks constructor to launch().
+    demo.queue().launch(theme=gr.themes.Citrus(), css=CSS, mcp_server=True)

examples/historic-interior.glb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d3609e9d0de9c1e6b7d99a60fd47dff832a4e727038b4b63f1639150a35c749a
+size 26084836