|
--- |
|
library_name: transformers |
|
tags: [] |
|
--- |
|
|
|
# Model Card for Model ID |
|
|
|
<!-- Provide a quick summary of what the model is/does. --> |
|
|
|
--- |
|
tags: |
|
- text-classification |
|
- sst2 |
|
- sentiment-analysis |
|
--- |
|
|
|
### Model Description |
|
This is a fine-tuned version of the `google/gemma-2b-it` model on the SST-2 dataset for sentiment classification. It predicts whether a sentence expresses positive or negative sentiment. |
|
|
|
### Example |
|
```python |
|
from transformers import pipeline |
|
|
|
classifier = pipeline("text-classification", model="ishanarang/gemma-sst2-small-finetuned") |
|
print(classifier("I love this movie!")) |
|
|
|
|
|
|
|
|
