# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("CouchCat/ma_ner_v6_distil")
model = AutoModelForTokenClassification.from_pretrained("CouchCat/ma_ner_v6_distil", device_map="auto")Quick Links
Description
A Named Entity Recognition model trained on a customer feedback data using DistilBert. Possible labels are:
- PRD: for certain products
- BRND: for brands
Usage
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("CouchCat/ma_ner_v6_distil")
model = AutoModelForTokenClassification.from_pretrained("CouchCat/ma_ner_v6_distil")
- Downloads last month
- 24
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="CouchCat/ma_ner_v6_distil")