Views
No views yet
1>>> from transformers import pipeline, TextGenerationPipeline, GPT2LMHeadModel, AutoTokenizer
2>>> model_name = "aspis/gpt2-genre-story-generation"
3>>> model = GPT2LMHeadModel.from_pretrained(model_name)
4>>> tokenizer = AutoTokenizer.from_pretrained(model_name)
5>>> generator = TextGenerationPipeline(model=model, tokenizer=tokenizer)
6# Input should be of format "<BOS> <Genre token> Optional starter text"
7>>> input_prompt = "<BOS> <adventure>"
8>>> story = generator(input_prompt, max_length=80, do_sample=True,
9 repetition_penalty=1.5, temperature=1.2,
10 top_p=0.95, top_k=50)
11>>> print(story)
12
13[{'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;'}]