a1hmad23/mva-call-classifier-v5

MVA Call Classifier (v5)

Inputs

Lowercase, ASR-style transcripts. Truncated to 128 tokens.

from transformers import DistilBertTokenizerFast, DistilBertForSequenceClassification
import torch

model_id = "a1hmad23/mva-call-classifier-v5"
tokenizer = DistilBertTokenizerFast.from_pretrained(model_id)
model = DistilBertForSequenceClassification.from_pretrained(model_id)
model.eval()

text = "yes i was in an accident last month"
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=128)
with torch.no_grad():
    logits = model(**inputs).logits
pred_id = logits.argmax(-1).item()
print(model.config.id2label[pred_id])

Labels

39 classes. The full mapping is in label2id.json and embedded in config.json. Label semantics, precedence rules, and confusable-neighbor decision rules are documented internally and are not redistributed with this model.