Update app.py

app.py

@@ -69,7 +69,7 @@ uploaded_file = st.file_uploader("Upload an Image", type=["jpg", "jpeg", "png"])
 if uploaded_file:
     # Load image
     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)
 
     # Sidebar Tools
     st.sidebar.title("Editing Tools")
@@ -126,7 +126,7 @@ if uploaded_file:
         st.error(f"Error during enhancement: {e}")
 
     # Display Enhanced Image
-    st.image(image, caption="Enhanced Image", use_column_width=True)
+    st.image(image, caption="Enhanced Image", use_container_width=True)
 
     # Download Button
     buf = io.BytesIO()