Model Card for Model ID
This model is developed to tag Names, Organisations and addresses. I have used a data combined fro Conll, ontonotes5, and a custom address dataset that was self made. Cleaned out the tags. Detects U.S addresses. ["O", "B-ORG", "I-ORG", "B-PER", "I-PER",'B-addr','I-addr']
Model Description
- Developed by: ctrlbuzz
- Model type: Bert
- Language(s) (NLP): Named Entity recognition
- Finetuned from model [optional]: bert-base-cased
Uses
Direct Use
from transformers import AutoTokenizer, AutoModelForTokenClassification
from transformers import pipeline
tokenizer = AutoTokenizer.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.",
print(nlp(example))
- Downloads last month
- 415
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.