Update README.md
Browse files
README.md
CHANGED
@@ -18,10 +18,10 @@ from PIL import Image
|
|
18 |
|
19 |
# 1) Load the fine-tuned model
|
20 |
model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(
|
21 |
-
"hf-hub:
|
22 |
pretrained=None
|
23 |
)
|
24 |
-
tokenizer = open_clip.get_tokenizer("hf-hub:
|
25 |
|
26 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
27 |
model.to(device)
|
@@ -68,7 +68,7 @@ import open_clip
|
|
68 |
from PIL import Image
|
69 |
|
70 |
model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(
|
71 |
-
"hf-hub:
|
72 |
pretrained=None
|
73 |
)
|
74 |
tokenizer = open_clip.get_tokenizer("hf-hub:your-username/OpenCLIP-BiomedCLIP-Finetuned")
|
|
|
18 |
|
19 |
# 1) Load the fine-tuned model
|
20 |
model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(
|
21 |
+
"hf-hub:mgbam/OpenCLIP-BiomedCLIP-Finetuned",
|
22 |
pretrained=None
|
23 |
)
|
24 |
+
tokenizer = open_clip.get_tokenizer("hf-hub:mgbam/OpenCLIP-BiomedCLIP-Finetuned")
|
25 |
|
26 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
27 |
model.to(device)
|
|
|
68 |
from PIL import Image
|
69 |
|
70 |
model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(
|
71 |
+
"hf-hub:mgbam/OpenCLIP-BiomedCLIP-Finetuned",
|
72 |
pretrained=None
|
73 |
)
|
74 |
tokenizer = open_clip.get_tokenizer("hf-hub:your-username/OpenCLIP-BiomedCLIP-Finetuned")
|