Update app.py

app.py

@@ -30,8 +30,8 @@ def text2story(text):
     Returns:
         str: Generated story suitable for kids aged 3-10.
     """
-    # Adjusted prompt to ensure the model generates a story
-    prompt = f"Write a short, happy, and fun story for kids aged 3-10 about: {text}. Make it cheerful, imaginative, and suitable for young children. Avoid any scary or sad elements. The story should be under 95 words."
+    # Simplified and rephrased prompt to guide the model
+    prompt = f"Write a short, happy story for kids about {text}. The story should be cheerful, imaginative, and under 95 words."
     
     text_generator = pipeline("text2text-generation", model="t5-small")
     story = text_generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']