Update app.py

app.py

@@ -168,9 +168,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")