titanabrian/whatsapp-text-classifier-distilbert

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