Upload app.py with huggingface_hub

app.py

@@ -10,7 +10,7 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 from huggingface_hub import snapshot_download, login
-import tempfile, uuid
+import tempfile
 from pathlib import Path
 
 if os.environ.get("HF_TOKEN"):
@@ -34,13 +34,12 @@ def _pip(*a):
         print(f"  pip OK: {tag}")
     else:
         print(f"  pip FAIL: {tag}")
-        print(f"    stderr: {r.stderr[-300:]}")
+        print(f"    {r.stderr[-300:]}")
     return ok
 
 print("=== Runtime installs ===")
 _pip("open3d>=0.18.0")
-# utils3d - pure Python wheel, install with verbose
-_pip("-v", "utils3d")
+_pip("utils3d")
 _pip("iopath")
 _pip("--no-deps", "sam2>=1.1.0")
 _pip("--no-deps", "git+https://github.com/microsoft/MoGe.git@a8c37341bc0325ca99b9d57981cc3bb2bd3e255b")
@@ -51,13 +50,8 @@ for idx in ["https://docs.gsplat.studio/whl/pt210cu128",
     if _pip("--no-deps", f"--extra-index-url={idx}", "gsplat"):
         break
 
-# Quick verify
-for mod in ["utils3d", "open3d", "gsplat", "kaolin", "pytorch3d"]:
-    try:
-        __import__(mod)
-        print(f"  import {mod}: OK")
-    except ImportError as e:
-        print(f"  import {mod}: FAIL - {e}")
+# DO NOT import CUDA-dependent packages here - only in GPU functions!
+# Importing gsplat/open3d at startup can init CUDA and break ZeroGPU.
 
 # --- Clone sam-3d-objects ---
 SAM3D_PATH = Path("/home/user/app/sam-3d-objects")
@@ -89,27 +83,38 @@ if hf_ckpt.exists() and not local_ckpt.exists():
     local_ckpt.symlink_to(hf_ckpt)
 CONFIG_PATH = str(local_ckpt / "pipeline.yaml")
 print(f"Config exists: {Path(CONFIG_PATH).exists()}")
-print("=== Startup complete ===")
+print("=== Startup complete (NO CUDA imports at startup) ===")
 
 # --- Endpoints ---
 
 @spaces.GPU(duration=60)
 def diagnose():
+    """Test GPU and modules in isolated manner."""
     import torch
     lines = [f"torch={torch.__version__}", f"cuda={torch.cuda.is_available()}"]
     if torch.cuda.is_available():
         lines.append(f"gpu={torch.cuda.get_device_name()}")
-    for mod in ["kaolin", "open3d", "utils3d", "iopath", "gsplat", "pytorch3d", "moge"]:
+    # Only test pure-python / safe imports
+    for mod in ["kaolin", "utils3d", "iopath", "pytorch3d"]:
+        try:
+            m = __import__(mod)
+            lines.append(f"{mod}: OK ({getattr(m, '__version__', '-')})")
+        except Exception as e:
+            lines.append(f"{mod}: FAIL - {e}")
+    # Test CUDA modules one at a time  
+    for mod in ["open3d", "gsplat", "moge"]:
         try:
             m = __import__(mod)
             lines.append(f"{mod}: OK ({getattr(m, '__version__', '-')})")
         except Exception as e:
             lines.append(f"{mod}: FAIL - {e}")
+    # sam2
     try:
         from sam2.automatic_mask_generator import SAM2AutomaticMaskGenerator
         lines.append("sam2: OK")
     except Exception as e:
         lines.append(f"sam2: FAIL - {e}")
+    # SAM3D inference
     try:
         from inference import Inference
         lines.append("SAM3D Inference: importable")
@@ -140,7 +145,7 @@ def reconstruct_objects(image: np.ndarray):
 
         preview = image_np.copy()
         preview[best_mask] = (preview[best_mask] * 0.5 + np.array([0, 255, 0]) * 0.5).astype(np.uint8)
-        print(f"  {len(masks)} masks, largest area={masks[0]['area']} ({time.time()-t0:.0f}s)")
+        print(f"  {len(masks)} masks ({time.time()-t0:.0f}s)")
 
         from inference import Inference
         sam3d = Inference(CONFIG_PATH, compile=False)
@@ -183,9 +188,6 @@ def reconstruct_objects(image: np.ndarray):
             m = result["mesh"]
             if hasattr(m, "export"):
                 m.export(glb)
-            else:
-                import open3d as o3d
-                o3d.io.write_triangle_mesh(glb, m)
         else:
             keys = list(result.keys()) if isinstance(result, dict) else dir(result)
             return None, preview, f"Cannot extract 3D. Keys: {keys}"