Downloads · 30 days
231
0% of all-time downloads
gpt2-genre-story-generator
pranavpsv/gpt2-genre-story-generator
gpt2-genre-story-generator is a text generation model from pranavpsv. Use it when you need the model to write or continue text. It is set up for transformers.
GPT2 fine-tuned on genre-based story generation.
All-time downloads
130K
Public
Repo size
1.5 GB
Likes
45
Public
Repo makeup
Click a slice to open those files.
.bin510 MB · 51%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- gpt2
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- May 23, 2021
- SHA
- d617d856
Try a prompt
Model card
From the Hugging Face model README
GPT2 Genre Based Story Generator
Model description
GPT2 fine-tuned on genre-based story generation.
Intended uses
Used to generate stories based on user inputted genre and starting prompts.
How to use
Supported Genres
superhero, action, drama, horror, thriller, sci_fi
Input text format
<BOS> <genre> Some optional text...
Example: <BOS> <sci_fi> After discovering time travel,
# Example of usage
from transformers import pipeline
story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
print(story_gen("<BOS> <superhero> Batman"))
Training data
Initialized with pre-trained weights of "gpt2" checkpoint. Fine-tuned the model on stories of various genres.
- Task
- Text Generation
- Library
- transformers
- Type
- gpt2
- Created
- Mar 2, 2022
- Updated
- May 23, 2021