Update README.md

README.md

@@ -80,9 +80,9 @@ The following hyperparameters were used during training:
 ### Code to Run
 
 def vit_classify(image):
-    # from transformers import ViTFeatureExtractor
-    # from transformers import ViTForImageClassification
-    # import torch
+    from transformers import ViTFeatureExtractor
+    from transformers import ViTForImageClassification
+    import torch
     
     vit = ViTForImageClassification.from_pretrained("oschamp/vit-artworkclassifier")
     vit.eval()