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