model updated

config.json

@@ -8,26 +8,14 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
-  "id2label": [
-    {
-      "0": "sadness"
-    },
-    {
-      "1": "joy"
-    },
-    {
-      "2": "love"
-    },
-    {
-      "3": "anger"
-    },
-    {
-      "4": "fear"
-    },
-    {
-      "5": "surprise"
-    }
-  ],
+  "id2label": {
+    "0": "sadness",
+    "1": "joy",
+    "2": "love",
+    "3": "anger",
+    "4": "fear",
+    "5": "surprise"
+  },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {

finetune-emotions.py

@@ -28,11 +28,12 @@ def get_encoded_data(tokenizer):
     return emotions_encoded
 
 if __name__ == "__main__":
-    labels = ['sadness', 'joy', 'love', 'anger', 'fear', 'surprise']
+    id2label = {0: 'sadness', 1: 'joy', 2: 'love', 3: 'anger', 4: 'fear', 5: 'surprise'}
+    labels = list(id2label.values())
     model = AutoModelForSequenceClassification.from_pretrained(
                 pretrained_model_name_or_path = BASE_MODEL_NAME,
                 num_labels = len(labels),
-                id2label=[{i: labels[i]} for i in range(len(labels))],
+                id2label = id2label,
                 resume_download=True,).to(device)
 
     tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL_NAME)

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8b0409f66d9c3fe0e3aa9976265fa1f26dbd1526cd212a696fa8d97e459b71e9
+oid sha256:416c31214cc859ad412b651c4bee2b72ca1dcb3801a53e780d4daeb5e126502f
 size 438036351