Update app.py

app.py

@@ -36,5 +36,9 @@ if uploaded_file is not None:
     # Compress the image using Truncated SVD
     compressed_image = truncated_svd_compress(resized_image, k)
 
+    # Normalize the images for display
+    gray_image_display = (resized_image - resized_image.min()) / (resized_image.max() - resized_image.min())
+    compressed_image_display = (compressed_image - compressed_image.min()) / (compressed_image.max() - compressed_image.min())
+
     # Display original and compressed images
-    st.image([gray_image, compressed_image], caption=["Original", "Compressed"])
+    st.image([gray_image_display, compressed_image_display], caption=["Original", "Compressed"], use_column_width=True)