import json
from pathlib import Path
from copy import copy

config = json.loads(Path("config.json").read_text())
code_map = json.loads(Path("code_map.json").read_text())  # label : ID
code_map_inv = {v: k for k, v in code_map.items()}  # id : label

code_map_inv_str = {str(k): str(v) for k, v in code_map_inv.items()}
code_map_str = {str(v): str(k) for k, v in code_map.items()}
config["id2label"] = code_map_inv_str
config["label2id"] = code_map

Path("config.json").write_text(json.dumps(config, indent=4))