Instructions to use dslim/distilbert-NER with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use dslim/distilbert-NER with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="dslim/distilbert-NER")# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("dslim/distilbert-NER") model = AutoModelForTokenClassification.from_pretrained("dslim/distilbert-NER") - Inference
- Notebooks
- Google Colab
- Kaggle
Fix token labels
#4
by ianroberts - opened
Fixes #3 - I inferred what I believe to be the correct mapping by passing some sample text with all the entity types to the inference API and observing the generated labels.
ianroberts changed pull request status to open
dslim changed pull request status to merged