Update src/streamlit_app.py

src/streamlit_app.py

@@ -33,9 +33,9 @@ if uploaded_file:
         # Display both images in columns
         col1, col2 = st.columns(2)
         with col1:
-            st.image(image, caption="Original Image", use_container_width=True)
+            st.image(image, caption="Original Image")
         with col2:
-            st.image(result_img, caption="Processed Image", use_container_width=True)
+            st.image(result_img, caption="Processed Image")
             
             
         # Create a buffer to store image bytes