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 vote241 views1 answer
How to make one list from nested list
Python
14 votes234 views4 answers
Plt subplots figsize
Python
0 vote234 views1 answer
Python save base64 temp file
Python
1 vote227 views1 answer
Django gmail smtp
Python
0 vote229 views2 answers
When excel is loaded into python, numeric datatype changes to float
Python
37 votes244 views6 answers
Lambda function in python
Python
1 vote235 views1 answer
Python get os cores
Python
0 vote227 views1 answer
Python find duplicated zip files
Python
2 votes263 views2 answers
Python selenium full screen
Python
10 votes232 views4 answers
Numpy empty array
Python