Update app.py

app.py

@@ -3,7 +3,6 @@ from PIL import Image
 # Load model directly
 from transformers import AutoProcessor, AutoModelForImageTextToText
 
-
 def generate_caption(image_path):
     image = Image.open(image_path)
     processor = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
@@ -41,7 +40,7 @@ def main():
     uploaded_file = st.file_uploader("选择一张图片", type=["png", "jpg", "jpeg"])
     if uploaded_file is not None:
         image = Image.open(uploaded_file)
-        st.image(image, caption="上传的图片", use_column_width=True)
+        st.image(image, caption="上传的图片", use_container_width=True)
         
         # 调用图像描述函数
         caption = generate_caption(uploaded_file)