Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 # function part
 # img2text
 def img2text(image_path):
-    image_to_text = pipeline("image-to-text", model="Maciel/Muge-Image-Caption")
+    image_to_text = pipeline("image-to-text", model="dumperize/movie-picture-captioning")
     text = image_to_text(image_path)[0]["generated_text"]
     return text
 
@@ -74,7 +74,7 @@ uploaded_file = st.file_uploader("Select an Image...")
 
 if uploaded_file is not None:
     # Display the uploaded image
-    st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
+    st.image(uploaded_file, caption="Uploaded Image", use_container_width=True)
     
     # Save the image temporarily
     image_path = save_uploaded_image(uploaded_file)