Upload app.py with huggingface_hub

app.py

@@ -89,7 +89,21 @@ def _build_evaluator() -> Any:
     return pipeline, evaluator
 
 
+def _prefetch_rembg() -> None:
+    """Download the u2net matting model at boot, outside any GPU window.
+
+    rembg fetches u2net.onnx (~170 MB) on its first session; doing that here
+    keeps the download out of every visitor's ZeroGPU quota.
+    """
+    try:
+        import rembg
+        rembg.new_session('u2net')
+    except Exception as error:  # noqa: BLE001 — optional path, never block boot
+        print(f'rembg prefetch skipped: {error}', flush=True)
+
+
 PIPELINE, EVALUATOR = _build_evaluator()
+_prefetch_rembg()
 CROP_RATIO = PIPELINE.preprocessing['crop_size_ratio']
 VIEW_LIMIT = int(PIPELINE.preprocessing['max_views'])
 
@@ -130,7 +144,7 @@ def _mesh_reprojections(mesh, extrinsics, intrinsics, resolution: int = 518) ->
     return images
 
 
-@spaces.GPU(duration=420)
+@spaces.GPU(duration=45)
 def reconstruct_views(images: List[str], seed: int = 1, remove_background: bool = False,
                       progress=gr.Progress(track_tqdm=True)) -> Tuple[str, List[str], str, str, str]:
     """Reconstruct a 3D object from 2-4 uncalibrated views of it.
@@ -218,7 +232,7 @@ CSS = """
 .dark .gradio-container { background: var(--background-fill-primary); }
 """
 
-with gr.Blocks(title='GenSfM — Generative Structure from Motion', theme=THEME, css=CSS) as demo:
+with gr.Blocks(title='GenSfM — Generative Structure from Motion') as demo:
     gr.Markdown(
         """
         # GenSfM — Generative Structure from Motion with Native 3D Diffusion
@@ -290,4 +304,4 @@ with gr.Blocks(title='GenSfM — Generative Structure from Motion', theme=THEME,
 
 if __name__ == '__main__':
     TMP_DIR.mkdir(parents=True, exist_ok=True)
-    demo.launch(mcp_server=True)
+    demo.launch(theme=THEME, css=CSS, mcp_server=True)