File size: 544 Bytes
86bc504
 
 
c19cf37
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
##Basic Usage


from transformers import BertTokenizer, AutoModelForTokenClassification
from transformers import pipeline

tokenizer = BertTokenizer.from_pretrained('bert-base-cased')
model = AutoModelForTokenClassification.from_pretrained("ctrlbuzz/bert-addresses")
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
example = "While Maria was representing Johnson & Associates at a conference in Spain, she mailed me a letter from her new office at 123 Elm St., Apt. 4B, Springfield, IL.",


ner_results = nlp(example)
print(ner_results)