Update config.json

config.json

@@ -9,6 +9,6 @@
     "vocab_size": 30522,
     "type_vocab_size": 2,
     "max_position_embeddings": 512,
-    "model_type": "image-classification"
+    "model_type": "vit"
 }