Views
No views yet
vinai/bartpho-syllable2565e-53210050.87510.83581from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4# Load model and tokenizer
5model_name = "visolex/bartpho-spam-binary"
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForSequenceClassification.from_pretrained(model_name)
8
9# Example review text
10text = "Sản phẩm này rất tốt, shop giao hàng nhanh!"
11
12# Tokenize
13inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=256)
14
15# Predict
16with torch.no_grad():
17 outputs = model(**inputs)
18 predicted_class = outputs.logits.argmax(dim=-1).item()
19 probabilities = torch.softmax(outputs.logits, dim=-1)
20
21
22# Map to label
23label_map = {0: "Non-spam", 1: "Spam"}
24predicted_label = label_map[predicted_class]
25confidence = probabilities[0][predicted_class].item()
26
27print(f"Text: {text}")
28print(f"Predicted: {predicted_label} (confidence: {confidence:.2%})")
291@misc{{
2 {model_key}_spam_detection,
3 title={{{description}}},
4 author={{ViSoLex Team}},
5 year={{2025}},
6 howpublished={{\url{{https://huggingface.co/{visolex/bartpho-spam-binary}}}}}
7}}