Update app.py

app.py

@@ -25,8 +25,7 @@ def generate_caption(image):
     return caption
 
 def generate_story(caption):
-    # Generate the story based on the caption using the GPT-2 model
-    prompt = f"Startig with 'Once upon a time...', also, based on the image described as '{caption}', here is a short, simple, and engaging story for children aged 3-10. The story should be easy to understand, use age-appropriate language, and convey a positive message. Focus on the main elements in the image and create a story that sparks their imagination:\n\n"
+    prompt = f"Once upon a time, based on the image described as '{caption}', here is a short, simple, and engaging story for children aged 3-10. The story should be easy to understand, use age-appropriate language, and convey a positive message. Focus on the main elements in the image and create a story that sparks their imagination:\n\n"
     story = story_generator(prompt, max_length=500, num_return_sequences=1)[0]["generated_text"]
     
     # Extract the story text from the generated output
@@ -36,7 +35,7 @@ def generate_story(caption):
     inappropriate_words = ["violence", "horror", "scary"]
     for word in inappropriate_words:
         story = story.replace(word, "")
-
+    
     # Limit the story to approximately 100 words
     words = story.split()
     if len(words) > 100: