use hf hub download

app.py

@@ -10,6 +10,7 @@ import torch
 from torchvision.transforms import transforms
 from torchvision.transforms import InterpolationMode
 import torchvision.transforms.functional as TF
+from huggingface_hub import hf_hub_download
 
 torch.set_grad_enabled(False)
 
@@ -137,7 +138,9 @@ class GatedHead(torch.nn.Module):
 
 model.head = GatedHead(min(model.head.weight.shape), 9083)
 
-safetensors.torch.load_model(model, "JTP_PILOT2-2-e3-vit_so400m_patch14_siglip_384.safetensors")
+hf_hub_download(repo_id="RedRocket/JointTaggerProject", subfolder="JTP_PILOT2", filename="JTP_PILOT2-e3-vit_so400m_patch14_siglip_384.safetensors")
+
+safetensors.torch.load_model(model, "JTP_PILOT2-e3-vit_so400m_patch14_siglip_384.safetensors")
 model.eval()
 
 with open("tagger_tags.json", "r") as file: