pytorch jax gpt2 lm-head causal-lm transformers text-generation
.gitattributes README.md added_tokens.json config.json eval_results_lm.txt flax_model.msgpack merges.txt pytorch_model.bin special_tokens_map.json tokenizer_config.json training_args.bin vocab.json

# 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,

```python
# 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.
