Views
No views yet
distilbert-base-multilingual-cased for intent classification of WhatsApp-style messages. It supports casual multilingual texts in English, Bahasa Indonesia, and Javanese.| Label | ID |
|---|---|
| Confirmation | 0 |
| Question | 1 |
| Task | 2 |
| Transactional | 3 |
| Unclassified | 4 |
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2
3model = AutoModelForSequenceClassification.from_pretrained("your-username/your-model-name")
4tokenizer = AutoTokenizer.from_pretrained("your-username/your-model-name")
5
6text = "Tolong kirim invoice hari ini."
7inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
8outputs = model(**inputs)
9prediction = outputs.logits.argmax(dim=-1).item()