fix(api): convert mask before blending source

ssube · Jan 15, 2023 · f142418 · f142418
1 parent 081a96d
commit f142418
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/image.py b/api/onnx_web/image.py
@@ -148,6 +148,6 @@ def expand_image(
 
     full_mask = mask_filter(mask_image, dims, origin)
     full_noise = noise_source(source_image, dims, origin)
-    full_source = Image.composite(full_source, full_noise, full_mask)
+    full_source = Image.composite(full_source, full_noise, full_mask.convert('L'))
 
     return (full_source, full_mask, full_noise, (full_width, full_height))