Spaces:

acozma
/

image_fft_playground

Runtime error

Andrei Cozma commited on Oct 31, 2023

Commit

0ee379e

1 Parent(s): 183418a

Updates

Files changed (1) hide show

app.py CHANGED Viewed

@@ -192,7 +192,7 @@ def onclick_process_fft(state, inp_image, mask_opacity, inverted_mask, pad):
         print("mask:", ImageInfo.from_any(image))
         image = Image.fromarray(image)
-        mask = Image.fromarray(mask)
         if not inverted_mask:
             mask = ImageOps.invert(mask)

         print("mask:", ImageInfo.from_any(image))
         image = Image.fromarray(image)
+        mask = Image.fromarray(mask).convert(image.mode)
         if not inverted_mask:
             mask = ImageOps.invert(mask)