update blend

app.py

@@ -4504,8 +4504,12 @@ with demo:
                 for i, ncut_image in enumerate(ncut_images):
                     original_image = original_images[i]
                     original_image = np.array(original_image)
+                    if original_image.dim == 2:
+                        original_image = np.stack([original_image] * 3, axis=-1)
                     ncut_image = ncut_image.resize(original_image.shape[:2][::-1], Image.LANCZOS)
                     ncut_image = np.array(ncut_image)
+                    from ncut_pytorch import convert_to_lab_color
+                    ncut_image = convert_to_lab_color(ncut_image)
                     blend_image = 0.5 * original_image + 0.5 * ncut_image
                     blend_image = blend_image.astype(np.uint8)
                     ncut_images[i] = Image.fromarray(blend_image)