Quotes Generator
Model description
This is a GPT2 model fine-tuned on the Quotes-500K dataset.
Intended uses & limitations
For a given user prompt, it can generate motivational quotes starting with it.
How to use
from transformers import AutoTokenizer, AutoModelWithLMHead
tokenizer = AutoTokenizer.from_pretrained("nandinib1999/quote-generator")
model = AutoModelWithLMHead.from_pretrained("nandinib1999/quote-generator")
Training data
This is the distribution of the total dataset into training, validation and test dataset for the fine-tuning task.
| train | 349796 |
|---|---|
| validation | 99942 |
| test | 49971 |
Training procedure
The model was fine-tuned using the Google Colab GPU for one epoch. The weights of the pre-trained GPT2 model were used as a base.
Eval results
| Epoch | Perplexity |
|---|---|
| 1 | 15.180 |
- Downloads last month
- 127
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.