YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
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.
- Downloads last month
- 496
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.