Downloads · 30 days
2
1% of all-time downloads
my-awesome-adapter
solwol/my-awesome-adapter
my-awesome-adapter is a text classification model from solwol. Use it when you need a label for a piece of text. It is set up for adapter-transformers.
An adapter for the roberta-base model that was trained on the sentiment/rottentomatoes dataset and includes a prediction head for classification.
All-time downloads
204
Public
Repo size
11.9 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin6 MB · 50%
At a glance
- Task
- Text Classification
- Library
- adapter-transformers
- Access
- Public
- Created
- Mar 26, 2024
- Updated
- Mar 27, 2024
- SHA
- 5011d78b
Try a prompt
Datasets
Model card
From the Hugging Face model README
Adapter solwol/my-awesome-adapter for roberta-base
An adapter for the roberta-base model that was trained on the sentiment/rotten_tomatoes dataset and includes a prediction head for classification.
This adapter was created for usage with the Adapters library.
Usage
First, install transformers and adapters:
pip install -U transformers adapters
Now, the adapter can be loaded and activated like this:
from adapters import AutoAdapterModel
model = AutoAdapterModel.from_pretrained("roberta-base")
adapter_name = model.load_adapter("solwol/my-awesome-adapter", source="hf", set_active=True)
adapter_name
Next, to perform sentiment classification:
from transformers import AutoTokenizer, TextClassificationPipeline
tokenizer = AutoTokenizer.from_pretrained("roberta-base")
classifier = TextClassificationPipeline(model=model, tokenizer=tokenizer)
classfifier("Adapters are awesome!")
- Task
- Text Classification
- Library
- adapter-transformers
- Created
- Mar 26, 2024
- Updated
- Mar 27, 2024