# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("kulkarni-harsh/address-extraction-ner")
model = AutoModelForTokenClassification.from_pretrained("kulkarni-harsh/address-extraction-ner", device_map="auto")Quick Links
Address Tagging model based on Distilbert Cased model trained using AWS Sagemaker.
- Downloads last month
- 1
Model tree for kulkarni-harsh/address-extraction-ner
Base model
distilbert/distilbert-base-cased
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="kulkarni-harsh/address-extraction-ner")