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 vote197 views1 answer
Install python on windows subsystem for linux
Python
7 votes201 views2 answers
Pygame keyboard input
Python
5 votes207 views2 answers
Django connexion session time
Python
6 votes189 views5 answers
Python fill zeros left
Python
0 vote218 views1 answer
Work day prior to date python
Python
0 vote203 views1 answer
Dataproc initialization_actions error
Python
0 vote270 views1 answer
Extract bounding boxes OpenCV
Python
0 vote212 views1 answer
Split the string using the default arguments
Python
1 vote217 views1 answer
Import CSV Files into R Using read_csv() method
Python
2 votes215 views2 answers
How to get percentage in python
Python