gpt2-poetry-manveer_2

GPT-2 Poetry Generator 🪶

This is a fine-tuned version of GPT-2 on a small dataset of 6 original English poem lines.
The model was trained to generate short poetic verses based on any given prompt.

📚 Dataset

Custom-created poems.txt file with 6 manually written poetic lines.
Each line represents a single poetic sentence.

🛠️ Training Details

• Model: GPT-2
• Epochs: 3
• Batch size: 2
• Learning rate: 5e-5
• Tokenizer: GPT-2 (with EOS token as pad)
• Platform: Google Colab using Trainer API
• Frameworks: 🤗 Transformers, Datasets

💡 How to Use

Try the model with the text-generation pipeline:

from transformers import pipeline

generator = pipeline("text-generation", model="YOUR_USERNAME/gpt2-poetry-manveer")
output = generator("In the shadow of the", max_new_tokens=50)
print(output[0]["generated_text"])