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

13 votes261 views4 answers
Python get screen size
Python
0 vote281 views1 answer
Flask print request headers
Python
0 vote211 views1 answer
Pyqt5 qpush button pressed
Python
0 vote242 views1 answer
Python change recursion depth
Python
0 vote271 views1 answer
Get linkinstance revit api
Python
4 votes238 views6 answers
If statement python
Python
0 vote242 views1 answer
Code converter C++ to python
Python
2 votes262 views1 answer
Python remove empty folders
Python
68 votes237 views5 answers
Pandas iterate rows
Python
-1 vote276 views1 answer
How to unstack multiindex pandas
Python