tsak6666/my_first_model

My First Model

This is a pre-trained DistilBERT model fine-tuned on the SST-2 dataset for sentiment analysis. It predicts whether a given text expresses positive or negative sentiment.

How to Use

from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("your_username/my-first-model")
tokenizer = AutoTokenizer.from_pretrained("your_username/my-first-model")
inputs = tokenizer("This movie is great!", return_tensors="pt")
outputs = model(**inputs)