Update app.py

app.py

@@ -503,7 +503,7 @@ with cols[0]:
     # Display uploaded image preview
     if uploaded_image:
         image_bytes_for_state = uploaded_image.getvalue()
-        st.image(image_bytes_for_state, caption="Your image", use_column_width='auto')
+        st.image(image_bytes_for_state, caption="Your image", use_container_width='auto')
         # Removed success message to save space
     else:
         st.info("Upload or take a picture.")
@@ -583,7 +583,7 @@ with cols[4]:
     st.subheader("3. Prediction Result") # H3 targeted by CSS
 
     if st.session_state.prediction_result and st.session_state.predicted_image_bytes:
-        st.image(st.session_state.predicted_image_bytes, caption="Image Analyzed", use_column_width='auto')
+        st.image(st.session_state.predicted_image_bytes, caption="Image Analyzed", use_container_width='auto')
 
         result_class = st.session_state.prediction_result
         probability = st.session_state.predicted_prob