Update app.py

app.py

@@ -138,9 +138,13 @@ class GatedHead(torch.nn.Module):
 
 model.head = GatedHead(min(model.head.weight.shape), 9083)
 
-hf_hub_download(repo_id="RedRocket/JointTaggerProject", subfolder="JTP_PILOT2", filename="JTP_PILOT2-e3-vit_so400m_patch14_siglip_384.safetensors")
+cached_model = 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")
+safetensors.torch.load_model(model, cached_model)
 model.eval()
 
 with open("tagger_tags.json", "r") as file: