Rajaram1996
/

FacialEmoRecog

Image Classification Transformers PyTorch

English vit image CLassification Inference Endpoints

Model card Files Files and versions Community

Rajaram Sivaramakrishnan commited on Nov 5, 2021

Commit

cd03909

•

1 Parent(s): 7e81b3d

Update config.json

Files changed (1) hide show

config.json +8 -8

config.json CHANGED Viewed

@@ -21,14 +21,14 @@
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "anger": "0",
-    "contempt": "1",
-    "disgust": "2",
-    "fear": "3",
-    "happy": "4",
-    "neutral": "5",
-    "sadness": "6",
-    "surprise": "7"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
+    "anger": 0,
+    "contempt": 1,
+    "disgust": 2,
+    "fear": 3,
+    "happy": 4,
+    "neutral": 5,
+    "sadness": 6,
+    "surprise": 7
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",