gh-lyrics-small-gpt / README.md
Ra-Is's picture
added the model base to the readme
99ccfe3
|
raw
history blame
689 Bytes
---
license: apache-2.0
pipeline_tag: text-generation
base_model: gpt2-medium
---
# Lyrics Generation Model
## Overview
This repository contains a fine-tuned GPT-2 model for generating song lyrics. The model was trained on a dataset of Ghanaian artist lyrics and is designed to generate coherent and contextually relevant lyrics based on input prompts.
## Model Details
- **Model Type**: GPT-2
- **Model Size**: medium
- **Tokenizer**: GPT-2 tokenizer
- **Training Data**: Lyrics from various Ghanaian artists
## Installation
To use this model, you'll need to install the required libraries. You can do this using `pip`:
```bash
pip install transformers
pip install tensorflow
```