lamsytan/sentiment-analysis-base-phobert

import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("lamsytan/sentiment-analysis-base-phobert")
model = AutoModelForSequenceClassification.from_pretrained("lamsytan/sentiment-analysis-base-phobert")

sentence = "Áo đẹp lắm nhá lần sau sẽ ghé tiếp ạ"

inputs = tokenizer(sentence, return_tensors="pt", padding=True, truncation=True)

with torch.no_grad():
    outputs = model(**inputs)
    logits = outputs.logits
    probabilities = torch.softmax(logits, dim=-1)

print(probabilities.tolist())
    # Output:
    # [[0.010827462188899517, 0.9538241624832153, 0.035348404198884964]]
    #         ^                         ^                  ^