Upload ViTForImageClassification

config.json

@@ -9,30 +9,23 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "sad",
-    "1": "disgust",
-    "2": "angry",
-    "3": "neutral",
-    "4": "fear",
-    "5": "surprise",
-    "6": "happy"
+    "0": "frowning",
+    "1": "kiss",
+    "2": "neutral",
+    "3": "openmouth",
+    "4": "raisedbrows",
+    "5": "smile"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "angry": 2,
-    "disgust": 1,
-    "fear": 4,
-    "frowning": 7,
-    "happy": 6,
-    "kiss": 8,
-    "neutral": 3,
-    "openmouth": 9,
-    "raisedbrows": 10,
-    "sad": 0,
-    "smile": 11,
-    "surprise": 5
+    "frowning": 0,
+    "kiss": 1,
+    "neutral": 2,
+    "openmouth": 3,
+    "raisedbrows": 4,
+    "smile": 5
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c3ea033cea51addaa9fdae55bbdf1142a7ae4f37ca2842fcfea7ca36dcaa05bd
-size 343239356
+oid sha256:ce2c6a350d1bab595f765d5af55abde3b9cd74310bf9b48d68c48381349ff51a
+size 343236280