app.py

app.py

@@ -20,6 +20,7 @@ st.title("Mask Your Own Inpaint")
 def load_images():
     """Loads images from an NPZ file only once to cache."""
     data = np.load("data/sflckr_all_images.npz")
+
     images = data["images"]
     return images
 
@@ -32,13 +33,6 @@ images = load_images()
 if st.button("Random Pick"):
     st.session_state.random_idx = random.randint(0, images.shape[0] - 1)
 
-def make_square(img, target_size=300):
-    """Pads and resizes the image to a square."""
-    size = max(img.size)
-    background = Image.new("RGB", (size, size), (255, 255, 255))
-    offset = ((size - img.size[0]) // 2, (size - img.size[1]) // 2)
-    background.paste(img, offset)
-    return background.resize((target_size, target_size))
 
 def run_inpainting(random_idx, mask_array, image_queue, sampling_queue):
     """Starts inpainting and sends images to the queue."""
@@ -48,7 +42,6 @@ def run_inpainting(random_idx, mask_array, image_queue, sampling_queue):
 if st.session_state.random_idx is not None:
     img_array = images[st.session_state.random_idx]
     img_pil = Image.fromarray(img_array)
-    img_pil = make_square(img_pil, target_size=512)  # Resized to 512x512
 
     # Set up drawing canvas and display columns
     col1, col2 = st.columns(2)
@@ -76,10 +69,13 @@ if st.session_state.random_idx is not None:
             st.write("Masked Image")
             st.image(binary_mask, caption="Binary Mask", width=300)
 
+
         mask_image = Image.fromarray(binary_mask)
         mask_array = 255 - np.array(mask_image)
         mask_array = np.expand_dims(mask_array, axis=(0, 1))
 
+        
+
         # Inpaint button action
         if st.button("inpaint"):
             st.write("Please wait...")