asmaabid/dreaddit-test
Viewer • Updated • 300 • 8
How to use jnyx74/stress-prediction with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="jnyx74/stress-prediction") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("jnyx74/stress-prediction")
model = AutoModelForSequenceClassification.from_pretrained("jnyx74/stress-prediction")This is a final year project\n Trained with dataset from kaggle.com\n Source:https://www.kaggle.com/datasets/ruchi798/stress-analysis-in-social-media