Update app.py

app.py

@@ -15,7 +15,7 @@ except ImportError:
     import torch
 
 # Load the image captioning model
-caption_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+caption_model = pipeline("image-to-text", model="unography/blip-large-long-cap")
 
 # Load the GPT-2 model for story generation
 story_generator = pipeline("text-generation", model="openai-community/gpt2")
@@ -27,8 +27,8 @@ 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 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"]
+    prompt = f"As a storyteller, please 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, and starting with wordings'one upon a time'"
+    story = story_generator(prompt, max_length=100, num_return_sequences=1)[0]["generated_text"]
     return story
 
 def convert_to_audio(story):