Downloads · 30 days
11
31% of all-time downloads
poem_Generator
mehwish67/poem_Generator
poem_Generator is a machine learning model from mehwish67. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
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 and the GPT-2 model from Huggin…
All-time downloads
35
Public
Parameters
124M
996 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors498 MB · 100%
At a glance
- License
- mit
- Model type
- gpt2
- Access
- Public
- Created
- Nov 7, 2024
- Updated
- Dec 9, 2024
- SHA
- 83cd74ff
Base models
Datasets
Model card
From the Hugging Face model README
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 and the GPT-2 model 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
- 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
- Fine-Tuned Model: This GPT-2 model has been fine-tuned specifically for poetry generation on the provided dataset.
- Type
- gpt2
- License
- mit
- Created
- Nov 7, 2024
- Updated
- Dec 9, 2024