fix(api): handle images with alpha channels better

ssube · Nov 25, 2023 · b1328fd · b1328fd
1 parent 192bd08
commit b1328fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/image/noise_source.py b/api/onnx_web/image/noise_source.py
@@ -94,7 +94,7 @@ def noise_source_normal(
 def noise_source_histogram(
     source: Image.Image, dims: Point, _origin: Point, **kw
 ) -> Image.Image:
-    r, g, b = source.split()
+    r, g, b, *_a = source.split()
     width, height = dims
     size = width * height