Hugging Face's logo

oschamp
/
vit-artworkclassifier

Image Classification
Transformers
PyTorch
TensorBoard
Safetensors
vit
Generated from Trainer
Eval Results (legacy)
Model card Files
xet
 Metrics Community
3
oschamp commited on
Commit
68ce7d5
·
1 Parent(s): c60a473

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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
+ ```