Update app.py

app.py

@@ -52,13 +52,12 @@ def main():
     
     if uploaded_image is not None:
         image = Image.open(uploaded_image)
-        st.image(image, caption="Uploaded Image", use_column_width=True)
-
+        st.image(image, caption="Uploaded Image", use_container_width=True)
         # Filters and enhancements
         filter_type = st.selectbox("Select Filter", ['None', 'Grayscale', 'Brightness'])
         if filter_type != 'None':
             image = apply_filter(image, filter_type)
-            st.image(image, caption="Filtered Image", use_column_width=True)
+            st.image(image, caption="Filtered Image", use_container_width=True)
         
         # Image adjustments
         brightness = st.slider("Brightness", 0.0, 2.0, 1.0)
@@ -82,7 +81,7 @@ def main():
         if flip_direction != 'None':
             image = flip_image(image, flip_direction)
         
-        st.image(image, caption="Edited Image", use_column_width=True)
+        st.image(image, caption="Edited Image", use_container_width=True)
         
         # Download the edited image
         if st.button("Download Image"):