Upload ViTForImageClassification

README.md

@@ -1,10 +1,10 @@
 ---
 license: apache-2.0
-base_model: google/vit-base-patch16-224-in21k
 tags:
 - generated_from_trainer
 metrics:
 - accuracy
+base_model: google/vit-base-patch16-224-in21k
 model-index:
 - name: snacks_classification
   results: []

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "_name_or_path": "/content/drive/MyDrive/Skill Academy/Project 8_Yayang Nurul Aulia Azizah/snacks-clf-model",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -11,6 +11,14 @@
   "id2label": {
     "0": "apple",
     "1": "banana",
+    "2": "cake",
+    "3": "candy",
+    "4": "carrot",
+    "5": "cookie",
+    "6": "doughnut",
+    "7": "grape",
+    "8": "hot dog",
+    "9": "ice cream",
     "10": "juice",
     "11": "muffin",
     "12": "orange",
@@ -20,15 +28,7 @@
     "16": "salad",
     "17": "strawberry",
     "18": "waffle",
-    "19": "watermelon",
-    "2": "cake",
-    "3": "candy",
-    "4": "carrot",
-    "5": "cookie",
-    "6": "doughnut",
-    "7": "grape",
-    "8": "hot dog",
-    "9": "ice cream"
+    "19": "watermelon"
   },
   "image_size": 224,
   "initializer_range": 0.02,