Update app.py

app.py

@@ -1,10 +1,9 @@
 import streamlit as st
 from PIL import Image
-import torch
 from transformers import pipeline
 
-# Initialize the Hugging Face model for image captioning
-image_captioning_model = pipeline("image-captioning")
+# Initialize the image captioning model
+image_captioning_model = pipeline("image-captioning", model="Salesforce/blip-image-captioning-large")
 
 # Streamlit app title and description
 st.title("Image Information Extractor")
@@ -27,8 +26,8 @@ if image:
     # Display the image description
     st.write("Image Description:")
     for caption in captions:
-        st.write(caption["caption"])
+        st.write(caption)
 
     # You can add more processing or explanations here if needed
 
-st.sidebar.text("Built with ❤️ by Your Moneeb")
+st.sidebar.text("Built with ❤️ by Your Name")