app.py

app.py

@@ -33,7 +33,6 @@ images = load_images()
 if st.button("Random Pick"):
     st.session_state.random_idx = random.randint(0, images.shape[0] - 1)
 
-
 def run_inpainting(random_idx, mask_array, image_queue, sampling_queue):
     """Starts inpainting and sends images to the queue."""
     vipainting.vipaint(random_idx, mask_array, image_queue, sampling_queue)
@@ -70,7 +69,8 @@ if st.session_state.random_idx is not None:
             st.image(binary_mask, caption="Binary Mask", width=300)
 
 
-        mask_image = Image.fromarray(binary_mask)
+        mask_image = Image.fromarray(binary_mask).resize((512, 512))  
+
         mask_array = 255 - np.array(mask_image)
         mask_array = np.expand_dims(mask_array, axis=(0, 1))