Update README.md
Browse files
README.md
CHANGED
@@ -85,7 +85,7 @@ The following hyperparameters were used during training:
|
|
85 |
- Tokenizers 0.13.2
|
86 |
|
87 |
### Code to Run
|
88 |
-
|
89 |
def vit_classify(image):
|
90 |
vit = ViTForImageClassification.from_pretrained("oschamp/vit-artworkclassifier")
|
91 |
vit.eval()
|
@@ -107,4 +107,4 @@ def vit_classify(image):
|
|
107 |
|
108 |
prediction = logits.argmax(-1)
|
109 |
return prediction.item() #vit.config.id2label[prediction.item()]
|
110 |
-
|
|
|
85 |
- Tokenizers 0.13.2
|
86 |
|
87 |
### Code to Run
|
88 |
+
```
|
89 |
def vit_classify(image):
|
90 |
vit = ViTForImageClassification.from_pretrained("oschamp/vit-artworkclassifier")
|
91 |
vit.eval()
|
|
|
107 |
|
108 |
prediction = logits.argmax(-1)
|
109 |
return prediction.item() #vit.config.id2label[prediction.item()]
|
110 |
+
```
|