Update app.py

app.py

@@ -10,16 +10,13 @@ def img2text(url):
     text = image_to_text_model(url)[0]["generated_text"]
     return text
 
-# Function to convert text to a kid-friendly story
+# Function to convert text to a story
 def text2story(text):
     # Initialize the text generation pipeline
     text_generator = pipeline("text-generation", model="distilbert/distilgpt2")
     
-    # Create a prompt to ensure the story is kid-friendly and happy
-    prompt = f"Write a happy and fun story for kids aged 3-10 based on the following scenario: {text}. The story should be suitable for children and should not contain any magical elements. Keep the story under 95 words."
-    
-    # Generate the story
-    story = text_generator(prompt, max_length=95, num_return_sequences=1)[0]['generated_text']
+    # Generate the story directly from the input text
+    story = text_generator(text, max_length=95, num_return_sequences=1)[0]['generated_text']
     
     return story