Update README.md
Browse files
README.md
CHANGED
@@ -14,7 +14,7 @@ metrics:
|
|
14 |
- accuracy
|
15 |
---
|
16 |
|
17 |
-
# Model card for vit_small_patch16_224.
|
18 |
|
19 |
A Vision Transformer (ViT) image classification model. \
|
20 |
Trained on 15M histology patches from PAIP and TCGA. \
|
@@ -51,7 +51,7 @@ img = Image.open(
|
|
51 |
|
52 |
# load model from the hub
|
53 |
model = timm.create_model(
|
54 |
-
model_name="hf-hub:1aurent/vit_small_patch16_224.
|
55 |
pretrained=True,
|
56 |
num_heads=12,
|
57 |
).eval()
|
|
|
14 |
- accuracy
|
15 |
---
|
16 |
|
17 |
+
# Model card for vit_small_patch16_224.transpath_mocov3
|
18 |
|
19 |
A Vision Transformer (ViT) image classification model. \
|
20 |
Trained on 15M histology patches from PAIP and TCGA. \
|
|
|
51 |
|
52 |
# load model from the hub
|
53 |
model = timm.create_model(
|
54 |
+
model_name="hf-hub:1aurent/vit_small_patch16_224.transpath_mocov3",
|
55 |
pretrained=True,
|
56 |
num_heads=12,
|
57 |
).eval()
|