Change label names on inference API

Assuming you uploaded your model using the instructions in the docs, my suggestion would be to copy all the contents in the folder produced by saving your model

model.save_pretrained(target_name_or_path)

to the repository where you original model is stored. Then you can run

cd path/to/original/model
git add . && git commit -m "Update labels"
git push

and that should update both config.json and pytorch_model.bin. If you don’t have the repository of the original model you can just clone it again

git lfs install
git clone https://huggingface.co/MoritzLaurer/policy-distilbert-7d

HTH!

1 Like