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
4,016
Inference Providers NEW
This model is not currently available via any of the supported third-party Inference Providers, and the model is not deployed on the HF Inference API.