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 vote275 views1 answer
Qcut and cut function in python stack overflow
Python
0 vote251 views1 answer
Delete nans in df python
Python
2 votes246 views2 answers
Instance variable python
Python
0 vote259 views1 answer
To create a SparkSession
Python
6 votes245 views2 answers
Jupyter display all columns
Python
8 votes258 views3 answers
How to maximize pandas output python
Python
0 vote273 views1 answer
"DjangoJSONEncoder" is not defined
Python Django
0 vote257 views1 answer
Django queryset to form
Python Django
1 vote243 views2 answers
Perfect number in python
Python
49 votes245 views6 answers
Sort dictionary
Python