Hugingface ner

from transformers import AutoTokenizer, AutoModelForTokenClassification
from transformers import pipeline

tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")

nlp = pipeline("ner", model=model, tokenizer=tokenizer)
example = "My name is Wolfgang and I live in Berlin"

ner_results = nlp(example)
print(ner_results)

Source: huggingface.co

Similar codes

0 vote300 views1 answer
Parcours de graph en largeur
Python
0 vote342 views1 answer
Python make an object hashable
Python
0 vote328 views1 answer
Legend of colorbar python
Python
5 votes312 views2 answers
Random boolean python
Python
2 votes315 views2 answers
Python url join
Python
1 vote330 views1 answer
Python print string name in pattern
Python
0 vote308 views1 answer
From string to flaot python numpy
Python
2 votes335 views1 answer
QTextEdit get text
Python
0 vote315 views1 answer
Write a line to a text file using the write() function
Python
21 votes334 views2 answers
Run python file using python code
Python