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")
|