Instructions to use ai4bharat/IndicNER with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ai4bharat/IndicNER with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="ai4bharat/IndicNER")# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("ai4bharat/IndicNER") model = AutoModelForTokenClassification.from_pretrained("ai4bharat/IndicNER") - Notebooks
- Google Colab
- Kaggle
what is the id2label mapping
#6
by archit11 - opened
i cant seem to find it anywhere
archit11 changed discussion status to closed
Hi, id2label can be found in the config here https://huggingface.co/ai4bharat/IndicNER/blob/main/config.json#L14 . This is responsible for converting the output indices to labels.
Thanks, I got the mapping with model.config.id2label