Add new models for image classification

vision_models.py

@@ -5,7 +5,10 @@ import numpy as np
 
 classification_models = {
     "ResNet18": "resnet_18_imagenet",
-    "ResNet50": "resnet_50_imagenet"
+    "ResNet50": "resnet_50_imagenet",
+    "ViT-B16-224": "vit_base_patch16_224_imagenet",
+    "ViT-B16-384": "vit_base_patch16_384_imagenet",
+    "ViT-L16-224": "vit_large_patch16_224_imagenet"
 }