Update app.py

app.py

@@ -171,9 +171,9 @@ if uploaded_files:
     with image_col:
         st.subheader("Processed Image")
         if 'processed_img' in st.session_state:
-            st.image(st.session_state.processed_img, use_column_width=True, caption="Processed Image")
+            st.image(st.session_state.processed_img, caption="Processed Image")
         else:
-            st.image(img, use_column_width=True, caption="Processed Image")
+            st.image(img, caption="Processed Image")
 
     with controls_col:
         st.subheader("Image Controls")
@@ -192,7 +192,7 @@ if uploaded_files:
 
     # Display Original Image Below
     st.subheader("Original Image")
-    st.image(img, use_column_width=True, caption="Original Image")
+    st.image(img, use_container_width=True, caption="Original Image")
 
     # Save and Export Options
     st.markdown("---")