Update app.py

app.py

@@ -18,7 +18,7 @@ pipe = pipeline("text-generation", model="openai-community/gpt2")
 
 def response(age, story):
     #messages = f"this is the 300 word story of {story} in words a {age} year old can understand."
-    messages = f"Tell the story of {story} in 300 words using language a {age} year old can understand."
+    messages = f"You are a school teacher telling stories to children. Retell the story of {story} in 300 words using language a {age} year old can understand."
     sequences = pipe(messages, max_length=500)
     gen_text = sequences[0]["generated_text"][len(messages):]
     return gen_text