bert-addresses / README.md
ctrlbuzz's picture
Update README.md
86bc504
|
raw
history blame
No virus
544 Bytes
##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)