Views
No views yet
1from transformers import pipeline
2
3classifier = pipeline(
4 "text-classification",
5 model="Talip7/bert-base-sst2-finetuned"
6)
7
8classifier("I love this project!")
91from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("Talip7/bert-base-sst2-finetuned")
5model = AutoModelForSequenceClassification.from_pretrained("Talip7/bert-base-sst2-finetuned")
6
7text = "This movie was absolutely fantastic!"
8
9inputs = tokenizer(text, return_tensors="pt")
10
11with torch.no_grad():
12 outputs = model(**inputs)
13
14logits = outputs.logits
15prediction = torch.argmax(logits, dim=-1).item()
16
17label_map = {0: "Negative", 1: "Positive"}
18print(f"Prediction: {label_map[prediction]}")
19