Update JTP_PILOT2/inference_gradio.py

JTP_PILOT2/inference_gradio.py

@@ -136,7 +136,7 @@ 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")
+safetensors.torch.load_model(model, "JTP_PILOT2-e3-vit_so400m_patch14_siglip_384.safetensors")
 
 if torch.cuda.is_available():
     model.cuda()