Update README.md

README.md

@@ -33,10 +33,10 @@ This can be used for text generation.
 >>> model_name = "aspis/gpt2-genre-story-generation"
 >>> model = GPT2LMHeadModel.from_pretrained(model_name)
 >>> tokenizer = AutoTokenizer.from_pretrained(model_name)
->>> story_generator = TextGenerationPipeline(model=model, tokenizer=tokenizer)
+>>> generator = TextGenerationPipeline(model=model, tokenizer=tokenizer)
 # Input should be of format "<BOS> <Genre token> Optional starter text"
->>> input_prompt = "<BOS> <adventure> He was trapped in the pirate's house"
->>> story = story_generator(input_prompt, max_length=80, do_sample=True,
+>>> 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)