Align label mapping with imdb dataset

config.json

@@ -24,5 +24,13 @@
   "transformers_version": "4.16.2",
   "type_vocab_size": 1,
   "use_cache": true,
-  "vocab_size": 50265
-}
+  "vocab_size": 50265,
+  "label2id": {
+    "neg": 0,
+    "pos": 1
+  },
+  "id2label": {
+    "0": "neg",
+    "1": "pos"
+  }
+}