Update app.py

app.py

@@ -59,7 +59,7 @@ if uploaded_file is not None:
     # Display the uploaded image
     image = Image.open(uploaded_file).convert("RGB")
     resized_image = image.resize((IMG_WIDTH, IMG_HEIGHT))
-    st.image(image, caption="Uploaded Image", use_column_width=True)
+    st.image(image, caption="Uploaded Image", use_container_width=True)
 
     # Predict the class
     with st.spinner("Predicting..."):