File size: 1,161 Bytes
b392137 6037bd5 8e8cbfc 9c06186 8e8cbfc |
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 |
---
base_model:
- openai-community/gpt2
datasets:
- merve/poetry
license: mit
---
# Poem Generation Project
Generate short, contextually relevant poems based on input themes or keywords (e.g., nature, love, seasons) using a fine-tuned GPT-2 model. This project uses the [Poetry Dataset by Merve](https://huggingface.co/datasets/merve/poetry) and the [GPT-2 model](https://huggingface.co/openai-community/gpt2) from Hugging Face.
## Project Overview
This project is designed to generate creative and stylistic short poems based on user-input themes or keywords. By fine-tuning a GPT-2 model on a curated poetry dataset, the model learns poetic structures and language styles, allowing it to generate verses relevant to a given theme.
## Dataset
- **Dataset**: [Poetry Dataset by Merve](https://huggingface.co/datasets/merve/poetry)
- This dataset contains various poems, which are used to fine-tune the GPT-2 model for poem generation.
## Model
- **Pre-trained Model**: [GPT-2 by OpenAI](https://huggingface.co/openai-community/gpt2)
- **Fine-Tuned Model**: This GPT-2 model has been fine-tuned specifically for poetry generation on the provided dataset.
|