How to set useful id2label and label2id in config.json using Trainer

# define mappings as dictionaries
id2label = {"0": "not toxic", "1": "toxic"}
label2id = {"not toxic": "0", "toxic": "1"}

# define config
config = AutoConfig.from_pretrained(model_name, label2id=label2id, id2label=id2label)
model = AutoModelForSequenceClassification.from_pretrained(model_name, config = config)