File size: 1,215 Bytes
104d4ea bf5ecd1 104d4ea 1a76289 104d4ea 1a76289 104d4ea 1a76289 104d4ea 1a76289 104d4ea 1a76289 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | ---
license: cc0-1.0
language:
- en
- fr
- tr
tags:
- art
- emoji
- brainrot
- text-generation
pretty_name: DistilGPT2 fine-tuned on YouTube Shorts comments
size_categories:
- 10M<n<100M
datasets:
- PingVortex/Youtube_shorts_comments
base_model:
- distilbert/distilgpt2
pipeline_tag: text-generation
library_name: transformers
---
# Youtube shorts comment generator
- Base model: [distilgpt2](https://huggingface.co/distilgpt2)
- Trained on: [YouTube Shorts Comments Dataset](https://huggingface.co/datasets/PingVortex/Youtube_shorts_comments)
## Model Details
- **Parameters**: 82M (DistilGPT2 architecture)
- **Training Data**: 1,475,500 YouTube Shorts comments
## Usage Example
```python
from transformers import pipeline
brainrot = pipeline('text-generation', model='PingVortex/Youtube-shorts-comment-generator')
output = brainrot("When you see a Sigma edit:", max_length=50)
print(output[0]['generated_text'])
```
*Sample output:*
`"When you see a Sigma edit: 😂😂😂😂 The white one on the last pic?😂😂😂😅😅😅😊😊😊😅😮😮😅"`
## Training Info
- **Epochs**: 1
- **Batch Size**: 8
- **Hardware**: Google Colab T4 GPU
- **Training Time**: ~2 hours
- **Loss**: 0.24 |