chaosbringerc/POSNEG
Viewer • Updated • 150 • 2
This model is trained on the POSNEG dataset for binary sentiment classification.
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
tokenizer = AutoTokenizer.from_pretrained("chaosbringerc/POSNEG-Dataset-API")
model = AutoModelForSequenceClassification.from_pretrained("chaosbringerc/POSNEG-Dataset-API")
text = "This is a great product!"
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)
predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)