Fix id2label mapping with actual CWE names

config.json

@@ -12,73 +12,73 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2",
-    "3": "LABEL_3",
-    "4": "LABEL_4",
-    "5": "LABEL_5",
-    "6": "LABEL_6",
-    "7": "LABEL_7",
-    "8": "LABEL_8",
-    "9": "LABEL_9",
-    "10": "LABEL_10",
-    "11": "LABEL_11",
-    "12": "LABEL_12",
-    "13": "LABEL_13",
-    "14": "LABEL_14",
-    "15": "LABEL_15",
-    "16": "LABEL_16",
-    "17": "LABEL_17",
-    "18": "LABEL_18",
-    "19": "LABEL_19",
-    "20": "LABEL_20",
-    "21": "LABEL_21",
-    "22": "LABEL_22",
-    "23": "LABEL_23",
-    "24": "LABEL_24",
-    "25": "LABEL_25",
-    "26": "LABEL_26",
-    "27": "LABEL_27",
-    "28": "LABEL_28",
-    "29": "LABEL_29",
-    "30": "LABEL_30"
+    "0": "safe",
+    "1": "CWE-20",
+    "2": "CWE-22",
+    "3": "CWE-78",
+    "4": "CWE-79",
+    "5": "CWE-89",
+    "6": "CWE-94",
+    "7": "CWE-119",
+    "8": "CWE-125",
+    "9": "CWE-190",
+    "10": "CWE-200",
+    "11": "CWE-264",
+    "12": "CWE-269",
+    "13": "CWE-276",
+    "14": "CWE-284",
+    "15": "CWE-287",
+    "16": "CWE-310",
+    "17": "CWE-327",
+    "18": "CWE-330",
+    "19": "CWE-352",
+    "20": "CWE-362",
+    "21": "CWE-399",
+    "22": "CWE-401",
+    "23": "CWE-416",
+    "24": "CWE-434",
+    "25": "CWE-476",
+    "26": "CWE-502",
+    "27": "CWE-601",
+    "28": "CWE-787",
+    "29": "CWE-798",
+    "30": "CWE-918"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "is_decoder": false,
   "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_10": 10,
-    "LABEL_11": 11,
-    "LABEL_12": 12,
-    "LABEL_13": 13,
-    "LABEL_14": 14,
-    "LABEL_15": 15,
-    "LABEL_16": 16,
-    "LABEL_17": 17,
-    "LABEL_18": 18,
-    "LABEL_19": 19,
-    "LABEL_2": 2,
-    "LABEL_20": 20,
-    "LABEL_21": 21,
-    "LABEL_22": 22,
-    "LABEL_23": 23,
-    "LABEL_24": 24,
-    "LABEL_25": 25,
-    "LABEL_26": 26,
-    "LABEL_27": 27,
-    "LABEL_28": 28,
-    "LABEL_29": 29,
-    "LABEL_3": 3,
-    "LABEL_30": 30,
-    "LABEL_4": 4,
-    "LABEL_5": 5,
-    "LABEL_6": 6,
-    "LABEL_7": 7,
-    "LABEL_8": 8,
-    "LABEL_9": 9
+    "safe": 0,
+    "CWE-20": 1,
+    "CWE-22": 2,
+    "CWE-78": 3,
+    "CWE-79": 4,
+    "CWE-89": 5,
+    "CWE-94": 6,
+    "CWE-119": 7,
+    "CWE-125": 8,
+    "CWE-190": 9,
+    "CWE-200": 10,
+    "CWE-264": 11,
+    "CWE-269": 12,
+    "CWE-276": 13,
+    "CWE-284": 14,
+    "CWE-287": 15,
+    "CWE-310": 16,
+    "CWE-327": 17,
+    "CWE-330": 18,
+    "CWE-352": 19,
+    "CWE-362": 20,
+    "CWE-399": 21,
+    "CWE-401": 22,
+    "CWE-416": 23,
+    "CWE-434": 24,
+    "CWE-476": 25,
+    "CWE-502": 26,
+    "CWE-601": 27,
+    "CWE-787": 28,
+    "CWE-798": 29,
+    "CWE-918": 30
   },
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,
@@ -92,4 +92,4 @@
   "type_vocab_size": 1,
   "use_cache": false,
   "vocab_size": 52000
-}
+}