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
Inference Examples
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.

Spaces using pranavpsv/gpt2-genre-story-generator 100