Update app.py

app.py

@@ -12,7 +12,7 @@ st.set_page_config(page_title="Storytelling Friend",
 uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])  # User uploads an image
 
 # Load the image captioning model
-caption_model = pipeline("image-captioning", model="Salesforce/blip-image-captioning-large")  # Load pre-trained model
+caption_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")  # Load pre-trained model
 
 # Load the text generation model
 story_model = pipeline("text-generation", model="gpt2")  # Load a text generation model