Update README.md

README.md

@@ -34,6 +34,7 @@ tokenizer = open_clip.get_tokenizer("ViT-L-14")
 import torch
 import open_clip
 from safetensors.torch import load_file
+from huggingface_hub import hf_hub_download
 from PIL import Image
 
 model_name = "ViT-L-14"
@@ -44,7 +45,14 @@ model, _, preprocess = open_clip.create_model_and_transforms(
     pretrained=None,
 )
 
-state_dict = load_file("open_clip_pytorch_model.safetensors")
+repo_id = "Yale-BIDS-Chen/medpmc-clip-l-14_jun24_v1"
+
+ckpt_path = hf_hub_download(
+    repo_id=repo_id,
+    filename="open_clip_pytorch_model.safetensors",
+)
+
+state_dict = load_file(ckpt_path, device="cpu")
 model.load_state_dict(state_dict, strict=True)
 model = model.to(device)
 model.eval()