finalized

gradio_app.py

@@ -12,6 +12,14 @@ from pathlib import Path
 import subprocess
 import uuid
 
+# --- HF_TOKEN INTEGRATION ---
+HF_TOKEN = os.environ.get("HF_TOKEN")
+if not HF_TOKEN:
+    raise ValueError(
+        "HF_TOKEN environment variable must be set to access gated models."
+    )
+# ----------------------------
+
 def install_cuda_toolkit():
     CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
     CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
@@ -43,11 +51,15 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 device = spar3d_utils.get_device()
 bg_remover = Remover()
+
+# --- HF_TOKEN PASSED HERE ---
 spar3d_model = SPAR3D.from_pretrained(
     "stabilityai/stable-point-aware-3d",
     config_name="config.yaml",
-    weight_name="model.safetensors"
+    weight_name="model.safetensors",
+    use_auth_token=HF_TOKEN  # Pass the token
 ).eval().to(device)
+# ----------------------------
 
 c2w_cond = spar3d_utils.default_cond_c2w(COND_DISTANCE)
 intrinsic, intrinsic_normed_cond = spar3d_utils.create_intrinsic_from_fov_rad(