Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("siridech24/sentiment")
5model = AutoModelForSequenceClassification.from_pretrained("siridech24/sentiment")
6
7inputs = tokenizer("ประทับใจบริการมากครับ", return_tensors="pt")
8outputs = model(**inputs)
9pred = torch.argmax(outputs.logits, dim=1)
10print(pred.item())
11