lhoestq/conll2003
Viewer • Updated • 20.7k • 2.72k • 5
How to use Abdirahman2040/ner-distilbert with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="Abdirahman2040/ner-distilbert") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("Abdirahman2040/ner-distilbert")
model = AutoModelForTokenClassification.from_pretrained("Abdirahman2040/ner-distilbert", device_map="auto")Base model
distilbert/distilbert-base-uncased