Hi @aadishhug, if I remember correctly, when I did these updates by hand in the config.json on the hub, the changes first did not show up in the inference widget, but then after a few days they did. Maybe the internal updates on the hub just take a while.
Otherwise, the solutions from @lewtun outlined above worked for me. the best is do input the label2id/id2label directly when you train/upload