Update app.py

app.py

@@ -18,7 +18,7 @@ except ImportError:
 caption_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
 
 # Load the GPT-2 model for story generation
-story_generator = pipeline("text-generation", model="gpt2")
+story_generator = pipeline("text-generation", model="openai-community/gpt2")
 
 def generate_caption(image):
     # Generate the caption for the uploaded image
@@ -27,7 +27,7 @@ def generate_caption(image):
 
 def generate_story(caption):
     # Generate the story based on the caption using the GPT-2 model
-    prompt = f"Imagine you are a storyteller for young children. Based on the image described as '{caption}', create a short and interesting story for children aged 3-10. Keep it positive and happy in tone."
+    prompt = f"Imagine you are a storyteller for young children. Based on the image described as '{caption}', create a interesting and attractive story for children aged 3-10. Keep it positive and happy in tone. Also, add more imagination"
     story = story_generator(prompt, max_length=200, num_return_sequences=1)[0]["generated_text"]
     return story