hf token auth

app.py

@@ -12,6 +12,8 @@ from PIL import Image
 from tsr.system import TSR
 from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
 
+HF_TOKEN = os.getenv("HF_TOKEN")
+
 if torch.cuda.is_available():
     device = "cuda:0"
 else:
@@ -21,6 +23,7 @@ model = TSR.from_pretrained(
     "stabilityai/TripoSR",
     config_name="config.yaml",
     weight_name="model.ckpt",
+    token=HF_TOKEN
 )
 model.to(device)
 

tsr/system.py

@@ -50,17 +50,17 @@ class TSR(BaseModule):
 
     @classmethod
     def from_pretrained(
-        cls, pretrained_model_name_or_path: str, config_name: str, weight_name: str
+        cls, pretrained_model_name_or_path: str, config_name: str, weight_name: str, token=None
     ):
         if os.path.isdir(pretrained_model_name_or_path):
             config_path = os.path.join(pretrained_model_name_or_path, config_name)
             weight_path = os.path.join(pretrained_model_name_or_path, weight_name)
         else:
             config_path = hf_hub_download(
-                repo_id=pretrained_model_name_or_path, filename=config_name
+                repo_id=pretrained_model_name_or_path, filename=config_name, token=token
             )
             weight_path = hf_hub_download(
-                repo_id=pretrained_model_name_or_path, filename=weight_name
+                repo_id=pretrained_model_name_or_path, filename=weight_name, token=token
             )
 
         cfg = OmegaConf.load(config_path)