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 vote116 views1 answer
Python how to reversetty.setraw(sys.stdin)
Python
4 votes110 views2 answers
Python code formatter vs code
Python
4 votes126 views7 answers
Int
Python
1 vote125 views1 answer
Handle queries in ListView django
Python Django
2 votes123 views1 answer
Termcolor python
Python
5 votes165 views2 answers
Copy directory from one location to another python
Python
1 vote115 views3 answers
Python delete file with extension
Python
7 votes107 views6 answers
Python calculator
Python
1 vote107 views2 answers
Deleting in a text file in python
Python
0 vote127 views1 answer
Mechanize python XE #27
Python