Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("mr-checker/sentimentX-distilbert")
5model = AutoModelForSequenceClassification.from_pretrained("mr-checker/sentimentX-distilbert")
6
7text = "SentimentX is working perfectly!"
8inputs = tokenizer(text, return_tensors="pt")
9
10with torch.no_grad():
11 logits = model(**inputs).logits
12
13predicted_class_id = logits.argmax().item()
14# 0: Negative, 1: Positive