WhatsApp Multilingual Intent Classifier
This model is a fine-tuned version of distilbert-base-multilingual-cased for intent classification of WhatsApp-style messages. It supports casual multilingual texts in English, Bahasa Indonesia, and Javanese.
The model classifies each message into one of five intent categories.
🏷️ Label Mapping
| Label | ID |
|---|---|
| Confirmation | 0 |
| Question | 1 |
| Task | 2 |
| Transactional | 3 |
| Unclassified | 4 |
📖 How to Use
from transformers import AutoTokenizer, AutoModelForSequenceClassification
model = AutoModelForSequenceClassification.from_pretrained("your-username/your-model-name")
tokenizer = AutoTokenizer.from_pretrained("your-username/your-model-name")
text = "Tolong kirim invoice hari ini."
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
outputs = model(**inputs)
prediction = outputs.logits.argmax(dim=-1).item()
- Downloads last month
- 6
Inference Providers
NEW
This model isn't deployed by any Inference Provider.
🙋
Ask for provider support