Update app.py

app.py

@@ -14,12 +14,12 @@ def text2story(text):
     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"
+        f"Please create a funny, heartwarming, imaginative fairy tale of no less than 50 words and less than 100 words."
+        f"Output only the body of the story, do not include other content."
     )
     
-    story_text = pipeline("text-generation", model="distilbert/distilgpt2",
-                              max_length=200, 
+    story_text = pipeline("text-generation", model="distilgpt2",
+                              max_length=100, 
                               do_sample=True, 
                               top_p=0.95, 
                               temperature=0.7)