Update app.py

app.py

@@ -1,12 +1,13 @@
 import streamlit as st
 from PIL import Image
-from transformers import AutoFeatureExtractor, AutoModelForImageCaptioning, AutoTokenizer
+import torch
+from transformers import AutoFeatureExtractor, AutoModelForCausalLM, AutoTokenizer
 
 # Load the image captioning model
 model_name = "Salesforce/blip-image-captioning-large"
 feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForImageCaptioning.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
 # Streamlit app title and description
 st.title("Image Information Extractor")