Update README.md

README.md

@@ -85,7 +85,7 @@ The following hyperparameters were used during training:
 - Tokenizers 0.13.2
 
 ### Code to Run
-'''
+```
 def vit_classify(image):
     vit = ViTForImageClassification.from_pretrained("oschamp/vit-artworkclassifier")
     vit.eval()
@@ -107,4 +107,4 @@ def vit_classify(image):
 
     prediction = logits.argmax(-1)
     return prediction.item() #vit.config.id2label[prediction.item()]
-'''
+```