CouchCat
/

ma_mlc_v7_distil

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Edit model card

Description

A Multi-label text classification model trained on a customer feedback data using DistilBert. Possible labels are:

Delivery (delivery status, time of arrival, etc.)
Return (return confirmation, return label requests, etc.)
Product (quality, complaint, etc.)
Monetary (pending transactions, refund, etc.)

Usage

from transformers import AutoTokenizer, AutoModelForSequenceClassification
  
  tokenizer = AutoTokenizer.from_pretrained("CouchCat/ma_mlc_v7_distil")
  
  model = AutoModelForSequenceClassification.from_pretrained("CouchCat/ma_mlc_v7_distil")

Downloads last month: 25

Inference API

Text 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.