
# GPT-2 fine-tuned for short story generation

Gpt-2 for short story generation with genres.

## Model description

Gpt-2 model fine-tuned on sample of BookCorpus dataset for short story generation, allows for the following genres (tokens to use as input under parenthesis):
- Romance (romance)
- Adventure (adventure)
- Mystery & detective (mystery-&-detective)
- Fantasy (fantasy)
- Humor & comedy (humor-&-comedy)
- Paranormal (paranormal)
- Science fiction (science-fiction)

Heavily inspired by https://huggingface.co/pranavpsv
## Intended uses & limitations

This can be used for text generation.

### How to use:

```python
>>> from transformers import pipeline, TextGenerationPipeline, GPT2LMHeadModel, AutoTokenizer
>>> model_name = "aspis/gpt2-genre-story-generation"
>>> model = GPT2LMHeadModel.from_pretrained(model_name)
>>> tokenizer = AutoTokenizer.from_pretrained(model_name)
>>> generator = TextGenerationPipeline(model=model, tokenizer=tokenizer)
# Input should be of format "<BOS> <Genre token> Optional starter text"
>>> input_prompt = "<BOS> <adventure>"
>>> story = generator(input_prompt, max_length=80, do_sample=True,
               repetition_penalty=1.5, temperature=1.2, 
               top_p=0.95, top_k=50)
>>> print(story)

[{'generated_text': '<BOS> <adventure> "How come they got that one?" asked Louran. The leader of the House, a young man with blonde hair and an odd grin...that didn\'t look so bad to her if she did have a smile on its face. She had known about this before. And now he\'d admitted it himself;'}]
```

## Training data

The model was trained using the BookCorpus dataset by getting the different genres per book and dividing the text into paragraphs.



{"language": ["en"], "license": "apache-2.0", "tags": ["text-generation"]}
transformers pytorch gpt2 text-generation en license:apache-2.0 autotrain_compatible endpoints_compatible text-generation-inference region:us
TAGS
#transformers #pytorch #gpt2 #text-generation #en #license-apache-2.0 #autotrain_compatible #endpoints_compatible #text-generation-inference #region-us 


# GPT-2 fine-tuned for short story generation

Gpt-2 for short story generation with genres.

## Model description

Gpt-2 model fine-tuned on sample of BookCorpus dataset for short story generation, allows for the following genres (tokens to use as input under parenthesis):
- Romance (romance)
- Adventure (adventure)
- Mystery & detective (mystery-&-detective)
- Fantasy (fantasy)
- Humor & comedy (humor-&-comedy)
- Paranormal (paranormal)
- Science fiction (science-fiction)

Heavily inspired by URL
## Intended uses & limitations

This can be used for text generation.

### How to use:



## Training data

The model was trained using the BookCorpus dataset by getting the different genres per book and dividing the text into paragraphs.



# GPT-2 fine-tuned for short story generation

Gpt-2 for short story generation with genres. ## Model description

Gpt-2 model fine-tuned on sample of BookCorpus dataset for short story generation, allows for the following genres (tokens to use as input under parenthesis):
- Romance (romance)
- Adventure (adventure)
- Mystery & detective (mystery-&-detective)
- Fantasy (fantasy)
- Humor & comedy (humor-&-comedy)
- Paranormal (paranormal)
- Science fiction (science-fiction)

Heavily inspired by URL ## Intended uses & limitations

This can be used for text generation. ### How to use: ## Training data

The model was trained using the BookCorpus dataset by getting the different genres per book and dividing the text into paragraphs.
TAGS
#transformers #pytorch #gpt2 #text-generation #en #license-apache-2.0 #autotrain_compatible #endpoints_compatible #text-generation-inference #region-us 
 # GPT-2 fine-tuned for short story generation

Gpt-2 for short story generation with genres. ## Model description

Gpt-2 model fine-tuned on sample of BookCorpus dataset for short story generation, allows for the following genres (tokens to use as input under parenthesis):
- Romance (romance)
- Adventure (adventure)
- Mystery & detective (mystery-&-detective)
- Fantasy (fantasy)
- Humor & comedy (humor-&-comedy)
- Paranormal (paranormal)
- Science fiction (science-fiction)

Heavily inspired by URL ## Intended uses & limitations

This can be used for text generation. ### How to use: ## Training data

The model was trained using the BookCorpus dataset by getting the different genres per book and dividing the text into paragraphs.
passage: TAGS
#transformers #pytorch #gpt2 #text-generation #en #license-apache-2.0 #autotrain_compatible #endpoints_compatible #text-generation-inference #region-us 
# GPT-2 fine-tuned for short story generation

Gpt-2 for short story generation with genres.## Model description

Gpt-2 model fine-tuned on sample of BookCorpus dataset for short story generation, allows for the following genres (tokens to use as input under parenthesis):
- Romance (romance)
- Adventure (adventure)
- Mystery & detective (mystery-&-detective)
- Fantasy (fantasy)
- Humor & comedy (humor-&-comedy)
- Paranormal (paranormal)
- Science fiction (science-fiction)

Heavily inspired by URL## Intended uses & limitations

This can be used for text generation.### How to use:## Training data

The model was trained using the BookCorpus dataset by getting the different genres per book and dividing the text into paragraphs.