Views
No views yet
vinai/phobert-base on the VN‑HSD dataset.vinai/phobert-base325e-5100256<INSERT_ACCURACY><INSERT_F1_SCORE>1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2
3tokenizer = AutoTokenizer.from_pretrained("visolex/phobert-hsd")
4model = AutoModelForSequenceClassification.from_pretrained("visolex/phobert-hsd")
5
6text = "Đừng nói những lời thô tục như vậy!"
7inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=256)
8pred = model(**inputs).logits.argmax(dim=-1).item()
9print(f"Label: {['CLEAN','OFFENSIVE','HATE'][pred]}")