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: 15

Inference API

Token Classification

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.