Update app.py

app.py

@@ -37,13 +37,13 @@ def text2story(text):
         str: Generated story suitable for kids aged 3-10, within 100 words.
     """
     # Add a prompt to ensure the story is happy, fun, and appropriate for kids
-    prompt = f"Base on the following scenario: {text}, write a happy and fun story for kids."
+    prompt = f"Write a happy and fun story for kids aged 3-10 based on the following scenario: {text}."
     
     # Load the text generation model
     story_generator = pipeline("text-generation", model="aspis/gpt2-genre-story-generation")
     
     # Generate the story
-    story = story_generator(prompt, max_length=150, num_return_sequences=1)[0]["generated_text"]
+    story = story_generator(prompt, max_length=150, num_return_sequences=1, temperature=0.7)[0]["generated_text"]
     
     # Remove the prompt from the generated story
     story = story.replace(prompt, "").strip()