可根据用户输入的类型（如超级英雄、科幻、恐怖等）和起始提示生成故事，支持多种类型，通过简单输入即可创作独特故事。【此简介由AI生成】

d617d856创建于 2021年5月23日32次提交

文件	最后提交记录	最后更新时间
.gitattributes	allow flax	4 年前
README.md	Migrate model card from transformers-repo Read announcement at https://discuss.huggingface.co/t/announcement-all-model-cards-will-be-migrated-to-hf-co-model-repos/2755 Original file history: https://github.com/huggingface/transformers/commits/master/model_cards/pranavpsv/gpt2-genre-story-generator/README.md	5 年前
added_tokens.json	Update added_tokens.json	5 年前
config.json	Update config.json	5 年前
eval_results_lm.txt	Update eval_results_lm.txt	5 年前
flax_model.msgpackLFS	upload flax model	4 年前
merges.txt	Update merges.txt	5 年前
pytorch_model.binLFS	Update pytorch_model.bin	5 年前
special_tokens_map.json	Update special_tokens_map.json	5 年前
tokenizer_config.json	Update tokenizer_config.json	5 年前
training_args.binLFS	Update training_args.bin	5 年前
vocab.json	Update vocab.json	5 年前

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"))