modified app.py

app.py

@@ -47,13 +47,13 @@ def main():
     if uploaded_files:
         for uploaded_file in uploaded_files:
             image = Image.open(uploaded_file)
-            st.image(image, caption=f"Original Image - {uploaded_file.name}", use_column_width=True)
+            st.image(image, caption=f"Original Image - {uploaded_file.name}", use_container_width=True)
             
             k = st.slider(f"Select number of colors (clusters) for {uploaded_file.name}", min_value=2, max_value=64, value=16)
             
             if st.button(f"Compress {uploaded_file.name}"):
                 compressed_img = compress_image_kmeans(image, k)
-                st.image(compressed_img, caption=f"Compressed Image - {uploaded_file.name}", use_column_width=True)
+                st.image(compressed_img, caption=f"Compressed Image - {uploaded_file.name}", use_container_width=True)
                 
                 st.markdown(get_image_download_link(compressed_img, filename=f"compressed_{uploaded_file.name}"), unsafe_allow_html=True)