Update app.py

app.py

@@ -10,8 +10,21 @@ def img2text(url):
     return text
 
 # text2story
+story_prompt = (
+    f"You are an expert in storytelling for children ages 3-10."
+    f"Based on the image description below:{text}\n"
+    f"Please create a funny, heartwarming, imaginative fairy tale of no less than 50 words"
+    f"Output only the body of the story, do not include other content"
+    )
+
 def text2story(text):
-    story_text = pipeline("text-generation", model="distilbert/distilgpt2")
+    story_text = pipeline("text-generation", model="distilbert/distilgpt2",
+                              max_length=200, 
+                              do_sample=True, 
+                              top_p=0.95, 
+                              temperature=0.7)
+    generation = story_text(story_prompt)
+    story_text = generation[0]['generated_text']
     return story_text
 
 # text2audio