Update config.json

config.json

@@ -2,7 +2,7 @@
   "_attn_implementation_autoset": true,
   "_name_or_path": "nielsr/vitpose-base-simple",
   "architectures": [
-    "ViTPoseForPoseEstimation"
+    "VitPoseForPoseEstimation"
   ],
   "backbone": null,
   "backbone_config": {
@@ -17,23 +17,23 @@
   "backbone_hidden_size": 768,
   "backbone_kwargs": null,
   "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"
+    "0": "Nose",
+    "1": "L_Eye",
+    "2": "R_Eye",
+    "3": "L_Ear",
+    "4": "R_Ear",
+    "5": "L_Shoulder",
+    "6": "R_Shoulder",
+    "7": "L_Elbow",
+    "8": "R_Elbow",
+    "9": "L_Wrist",
+    "10": "R_Wrist",
+    "11": "L_Hip",
+    "12": "R_Hip",
+    "13": "L_Knee",
+    "14": "R_Knee",
+    "15": "L_Ankle",
+    "16": "R_Ankle"
   },
   "image_size": [
     256,
@@ -41,23 +41,24 @@
   ],
   "initializer_range": 0.02,
   "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_2": 2,
-    "LABEL_3": 3,
-    "LABEL_4": 4,
-    "LABEL_5": 5,
-    "LABEL_6": 6,
-    "LABEL_7": 7,
-    "LABEL_8": 8,
-    "LABEL_9": 9
+    "Nose": "0",
+    "L_Eye": "1",
+    "R_Eye": "2",
+    "L_Ear": "3",
+    "R_Ear": "4",
+    "L_Shoulder": "5",
+    "R_Shoulder": "6",
+    "L_Elbow": "7",
+    "R_Elbow": "8",
+    "L_Wrist": "9",
+    "R_Wrist": "10",
+    "L_Hip": "11",
+    "R_Hip": "12",
+    "L_Knee": "13",
+    "R_Knee": "14",
+    "L_Ankle": "15",
+    "R_Ankle": "16"
+}
   },
   "model_type": "vitpose",
   "patch_size": [