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")