Views
No views yet
1from transformers import AutoTokenizer, AutoModelForTokenClassification
2from transformers import pipeline
3tokenizer = AutoTokenizer.from_pretrained('bert-base-cased')
4model = AutoModelForTokenClassification.from_pretrained("ctrlbuzz/bert-addresses")
5nlp = pipeline("ner", model=model, tokenizer=tokenizer)
6example = "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.",
7
8print(nlp(example))