Update app.py

app.py

@@ -36,24 +36,13 @@ def text2story(text):
     Returns:
         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"Write a kids story based on below scenario :{text}."
-    
+
     # Load the text generation model
     story_generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator")
     
     # Generate the story
-    story = story_generator(prompt, max_length=150, num_return_sequences=1)[0]["generated_text"]
-    
-    # Remove the prompt from the generated story
-    story = story.replace(prompt, "").strip()
-    
-    # Ensure the story is within 100 words
-    story_words = story.split()
-    if len(story_words) > 100:
-        story = " ".join(story_words[:100])
+    story = story_generator(f"<BOS> <superhero> {text}"))
     
-    return story
 
 # Function to convert text to audio using gTTS
 def text2audio(story_text):