Update pages/4_Image Augmentation.py

pages/4_Image Augmentation.py

@@ -39,7 +39,7 @@ def get_image_download_link(img_array, filename, text):
 uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
 if uploaded_file is not None:
     image = Image.open(uploaded_file)
-    st.image(image, caption='Uploaded Image.', use_column_width=True)
+    st.image(image, caption='Uploaded Image.', use_container_width=True)
     st.write("")
     st.write("Select the augmentation operations you want to apply:")
 
@@ -96,9 +96,9 @@ if uploaded_file is not None:
     if convert_gray:
         img_array_gray = cv2.cvtColor(img_array, cv2.COLOR_RGB2GRAY)
         img_array_gray = cv2.cvtColor(img_array_gray, cv2.COLOR_GRAY2RGB)  # Convert back to RGB for consistency
-        st.image(img_array_gray, caption='Grayscale Image.', use_column_width=True)
+        st.image(img_array_gray, caption='Grayscale Image.', use_container_width=True)
         st.markdown(get_image_download_link(img_array_gray, "grayscale.jpg", "Download Grayscale Image"), unsafe_allow_html=True)
 
-    # Convert numpy array back to image 
-        augmented_image = Image.fromarray(img_array) 
-        st.image(augmented_image, caption='Augmented Image.', use_column_width=True)
+    # Convert numpy array back to image
+    augmented_image = Image.fromarray(img_array)
+    st.image(augmented_image, caption='Augmented Image.', use_container_width=True)