Update config.json

config.json

@@ -8,121 +8,122 @@
   "dim": 768,
   "dropout": 0.1,
   "hidden_dim": 3072,
-  "id2label": {'0': 'AIR COMPRESSOR',
- "0" : "AIR COMPRESSOR" ,
-"1" : "ALLEN KEY" ,
-"2" : "ANGLE GAUGE" ,
-"3" : "AXE" ,
-"4" : "BACK SAW" ,
-"5" : "BAND SAW" ,
-"6" : "BONE SAW" ,
-"7" : "CALIPER" ,
-"8" : "CHISEL" ,
-"9" : "CIRCULAR SAW" ,
-"10" : "CLAMP" ,
-"11" : "COPING SAW" ,
-"12" : "CORDLESS DRILL" ,
-"13" : "CROSSCUT SAW" ,
-"14" : "DRILLS" ,
-"15" : "FILE" ,
-"16" : "FRET SAW" ,
-"17" : "GARAGE CABINET" ,
-"18" : "GAUGE" ,
-"19" : "GRINDER" ,
-"20" : "HACK SAW" ,
-"21" : "HAMMER" ,
-"22" : "HAND SAW" ,
-"23" : "HEAT GUN" ,
-"24" : "IMPACT DRIVER" ,
-"25" : "IMPACT WRENCH" ,
-"26" : "INCLINOMETER" ,
-"27" : "KEYHOLE SAW" ,
-"28" : "LASER LEVEL" ,
-"29" : "LATHE" ,
-"30" : "LEVEL" ,
-"31" : "MALLET" ,
-"32" : "MICROMETER" ,
-"33" : "MITER SAW" ,
-"34" : "NAIL GUN" ,
-"35" : "NUT DRIVER" ,
-"36" : "PLANE" ,
-"37" : "PLIERS" ,
-"38" : "PROTRACTOR" ,
-"39" : "PRUNING SAW" ,
-"40" : "RASP" ,
-"41" : "RECIPROCATING SAW" ,
-"42" : "ROLLING TOOL STORAGE" ,
-"43" : "SANDER" ,
-"44" : "SCREW DRIVERS" ,
-"45" : "STAPLER" ,
-"46" : "TABLE SAW" ,
-"47" : "TAPE MEASURE" ,
-"48" : "TILE SAW" ,
-"49" : "TOOL BAG" ,
-"50" : "TOOL BELT" ,
-"51" : "TOOL BOX" ,
-"52" : "TOOL CABINET" ,
-"53" : "TOOL CHEST" ,
-"54" : "VENEER SAW" ,
-"55" : "WRENCH"},
+  "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",
+    "31": "LABEL_31",
+    "32": "LABEL_32",
+    "33": "LABEL_33",
+    "34": "LABEL_34",
+    "35": "LABEL_35",
+    "36": "LABEL_36",
+    "37": "LABEL_37",
+    "38": "LABEL_38",
+    "39": "LABEL_39",
+    "40": "LABEL_40",
+    "41": "LABEL_41",
+    "42": "LABEL_42",
+    "43": "LABEL_43",
+    "44": "LABEL_44",
+    "45": "LABEL_45",
+    "46": "LABEL_46",
+    "47": "LABEL_47",
+    "48": "LABEL_48",
+    "49": "LABEL_49",
+    "50": "LABEL_50",
+    "51": "LABEL_51",
+    "52": "LABEL_52",
+    "53": "LABEL_53",
+    "54": "LABEL_54",
+    "55": "LABEL_55"
+  },
   "initializer_range": 0.02,
   "label2id": {
-    "AIR COMPRESSOR" : "0" ,
-"ALLEN KEY" : "1" ,
-"ANGLE GAUGE" : "2" ,
-"AXE" : "3" ,
-"BACK SAW" : "4" ,
-"BAND SAW" : "5" ,
-"BONE SAW" : "6" ,
-"CALIPER" : "7" ,
-"CHISEL" : "8" ,
-"CIRCULAR SAW" : "9" ,
-"CLAMP" : "10" ,
-"COPING SAW" : "11" ,
-"CORDLESS DRILL" : "12" ,
-"CROSSCUT SAW" : "13" ,
-"DRILLS" : "14" ,
-"FILE" : "15" ,
-"FRET SAW" : "16" ,
-"GARAGE CABINET" : "17" ,
-"GAUGE" : "18" ,
-"GRINDER" : "19" ,
-"HACK SAW" : "20" ,
-"HAMMER" : "21" ,
-"HAND SAW" : "22" ,
-"HEAT GUN" : "23" ,
-"IMPACT DRIVER" : "24" ,
-"IMPACT WRENCH" : "25" ,
-"INCLINOMETER" : "26" ,
-"KEYHOLE SAW" : "27" ,
-"LASER LEVEL" : "28" ,
-"LATHE" : "29" ,
-"LEVEL" : "30" ,
-"MALLET" : "31" ,
-"MICROMETER" : "32" ,
-"MITER SAW" : "33" ,
-"NAIL GUN" : "34" ,
-"NUT DRIVER" : "35" ,
-"PLANE" : "36" ,
-"PLIERS" : "37" ,
-"PROTRACTOR" : "38" ,
-"PRUNING SAW" : "39" ,
-"RASP" : "40" ,
-"RECIPROCATING SAW" : "41" ,
-"ROLLING TOOL STORAGE" : "42" ,
-"SANDER" : "43" ,
-"SCREW DRIVERS" : "44" ,
-"STAPLER" : "45" ,
-"TABLE SAW" : "46" ,
-"TAPE MEASURE" : "47" ,
-"TILE SAW" : "48" ,
-"TOOL BAG" : "49" ,
-"TOOL BELT" : "50" ,
-"TOOL BOX" : "51" ,
-"TOOL CABINET" : "52" ,
-"TOOL CHEST" : "53" ,
-"VENEER SAW" : "54" ,
-"WRENCH" : "55" 
+    "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_31": 31,
+    "LABEL_32": 32,
+    "LABEL_33": 33,
+    "LABEL_34": 34,
+    "LABEL_35": 35,
+    "LABEL_36": 36,
+    "LABEL_37": 37,
+    "LABEL_38": 38,
+    "LABEL_39": 39,
+    "LABEL_4": 4,
+    "LABEL_40": 40,
+    "LABEL_41": 41,
+    "LABEL_42": 42,
+    "LABEL_43": 43,
+    "LABEL_44": 44,
+    "LABEL_45": 45,
+    "LABEL_46": 46,
+    "LABEL_47": 47,
+    "LABEL_48": 48,
+    "LABEL_49": 49,
+    "LABEL_5": 5,
+    "LABEL_50": 50,
+    "LABEL_51": 51,
+    "LABEL_52": 52,
+    "LABEL_53": 53,
+    "LABEL_54": 54,
+    "LABEL_55": 55,
+    "LABEL_6": 6,
+    "LABEL_7": 7,
+    "LABEL_8": 8,
+    "LABEL_9": 9
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
@@ -137,4 +138,4 @@
   "torch_dtype": "float32",
   "transformers_version": "4.40.0",
   "vocab_size": 30522
-}
+}