Downloads · 30 days
8
6% of all-time downloads
my_first_model
tsak6666/my_first_model
my_first_model is a text classification model from tsak6666. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
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.
All-time downloads
127
Public
Repo size
268 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin268 MB · 100%
At a glance
- Task
- Text Classification
- License
- apache-2.0
- Model type
- distilbert
- Access
- Public
- Created
- Apr 17, 2025
- Updated
- Apr 17, 2025
- SHA
- c01d6e00
Try a prompt
Datasets
Model card
From the Hugging Face model README
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)
- Task
- Text Classification
- Type
- distilbert
- License
- apache-2.0
- Languages
- en
- Created
- Apr 17, 2025
- Updated
- Apr 17, 2025