Update app.py

app.py

@@ -27,7 +27,7 @@ except ImportError:
 # Load the image captioning model
 caption_model = pipeline("image-to-text", model="unography/blip-large-long-cap")
 
-story_generator = pipeline("text-generation", model="mosaicml/mpt-7b-storywriter", trust_remote_code=True)
+story_generator = pipeline("text-generation", model="ajibawa-2023/Young-Children-Storyteller-Mistral-7B")
 
 def generate_caption(image):
     # Generate the caption for the uploaded image