Fine-tuned version of
Simple_VieQA
for Vietnamese sentiment analysis on
UIT-VSFC dataset.
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4model_name = "ZycckZ/Simple_VieQA-sentiment"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForSequenceClassification.from_pretrained(model_name)
7
8text = "Sản phẩm rất tốt, pin trâu"
9inputs = tokenizer(text, return_tensors="pt")
10with torch.no_grad():
11 logits = model(**inputs).logits
12pred = torch.argmax(logits, dim=-1).item()
13print(pred)
