Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4model_name = "mdjrjoy/distilbert-my-task"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForSequenceClassification.from_pretrained(model_name)
8
9text = "The product quality is excellent."
10
11inputs = tokenizer(text, return_tensors="pt", truncation=True)
12
13with torch.no_grad():
14 outputs = model(**inputs)
15
16prediction = torch.argm**(outputs.logits, dim=1).item()
17
18l**els = {
19 0: "Negative",
20 1:**Neutral",
21 2: "Positive"
22}
23
24pr**t(labels[prediction])