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)