Update app.py

app.py

@@ -1,8 +1,12 @@
 import numpy as np
 import gradio as gr
-from skimage import io, color
-from numpy.linalg import norm
+from skimage import color
 from PIL import Image
+from skimage.transform import resize
+
+def resize_image(image, target_shape=(500, 500)):
+    """Resize image to target shape."""
+    return resize(image, target_shape, anti_aliasing=True)
 
 def svd_compress(image, k):
     """Compress the image using SVD by keeping only the top k singular values."""
@@ -18,8 +22,11 @@ def process_image(image, k):
     # Convert to grayscale
     gray_image = color.rgb2gray(image_np)
     
+    # Resize image to 500x500 for faster processing
+    resized_image = resize_image(gray_image, target_shape=(500, 500))
+    
     # Compress the image
-    compressed_image = svd_compress(gray_image, k)
+    compressed_image = svd_compress(resized_image, k)
     
     # Convert compressed image back to PIL Image for Gradio output
     compressed_image_pil = Image.fromarray((compressed_image * 255).astype(np.uint8))
@@ -28,10 +35,9 @@ def process_image(image, k):
 
 # Gradio interface
 gr.Interface(fn=process_image,
-             inputs=[gr.Image(type="pil", shape=(500, 500)), 
+             inputs=[gr.Image(type="pil"), 
                      gr.Slider(1, 100, step=1, value=50, label="Compression Rank")],
              outputs=gr.Image(type="pil"),
              title="Interactive Image Compression using SVD",
-             description="Upload an image (500x500 max) and adjust the compression rank.")
-
-gr_interface.launch()
+             description="Upload an image and adjust the compression rank to see the compressed version."
+).launch()