fix grey

app.py

@@ -4512,6 +4512,8 @@ with demo:
             plt.style.use('dark_background')
             
             def pad_image_with_border(image, border_color, border_width):
+                if len(image.shape) == 2:
+                    image = np.stack([image] * 3, axis=-1)
                 new_image = np.ones((image.shape[0] + 2 * border_width, image.shape[1] + 2 * border_width, image.shape[2]), dtype=image.dtype)
                 new_image[:, :] = border_color
                 new_image[border_width:-border_width, border_width:-border_width] = image