Align label mapping with imdb dataset

config.json

@@ -21,5 +21,13 @@
   "tie_weights_": true,
   "torch_dtype": "float32",
   "transformers_version": "4.18.0",
-  "vocab_size": 30522
-}
+  "vocab_size": 30522,
+  "label2id": {
+    "neg": 0,
+    "pos": 1
+  },
+  "id2label": {
+    "0": "neg",
+    "1": "pos"
+  }
+}