Update README.md

README.md

@@ -26,7 +26,7 @@ It's fine tuned on CiFAR10 for 1000 steps, and achieved accuracy of 98.7% on tes
 
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
-```
+```python
 from transformers import ViTImageProcessor, ViTForImageClassification
 from torchvision import datasets