Skip to content

Commit a7286ce

Browse files
author
hfhoffman1144
committed
first commit
1 parent 2bf9eb3 commit a7286ce

File tree

6 files changed

+10941
-0
lines changed

6 files changed

+10941
-0
lines changed
Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
import torch
2+
from transformers import (
3+
AutoTokenizer,
4+
AutoModelForSequenceClassification,
5+
pipeline,
6+
)
7+
8+
model_name = "cardiffnlp/twitter-roberta-base-sentiment-latest"
9+
10+
tokenizer = AutoTokenizer.from_pretrained(model_name)
11+
model = AutoModelForSequenceClassification.from_pretrained(model_name)
12+
13+
text = "I love using the Transformers library!"
14+
encoded_input = tokenizer(text, return_tensors="pt")
15+
16+
with torch.no_grad():
17+
output = model(**encoded_input)
18+
19+
scores = output.logits[0]
20+
probabilities = torch.softmax(scores, dim=0)
21+
predicted_class = probabilities.argmax().item()
22+
23+
print(f"Predicted class: {predicted_class}")
24+
print(f"Probabilities: {probabilities.tolist()}")
25+
26+
full_pipeline = pipeline(model=model_name)
27+
print(full_pipeline(text))

0 commit comments

Comments
 (0)