Update config.json

config.json

@@ -1,7 +1,7 @@
 {
   "_name_or_path": "bert-base-uncased",
   "architectures": [
-    "BertForTokenClassification"
+    "BertForSequenceClassification"
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
@@ -9,30 +9,8 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
-  "id2label": {
-    "0": "PERSON",
-    "1": "ORG",
-    "2": "LOC",
-    "3": "DATE",
-    "4": "TIME",
-    "5": "MONEY",
-    "6": "PERCENT",
-    "7": "EVENT",
-    "8": "OTHER"
-  },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
-  "label2id": {
-    "PERSON": 0,
-    "ORG": 1,
-    "LOC": 2,
-    "DATE": 3,
-    "TIME": 4,
-    "MONEY": 5,
-    "PERCENT": 6,
-    "EVENT": 7,
-    "OTHER": 8
-  },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",
@@ -40,6 +18,7 @@
   "num_hidden_layers": 12,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
+  "problem_type": "single_label_classification",
   "torch_dtype": "float32",
   "transformers_version": "4.39.3",
   "type_vocab_size": 2,