From b1328fdfdb747cd563762a91d9bbf0c9a37cf2e2 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 25 Nov 2023 07:50:54 -0600 Subject: [PATCH] fix(api): handle images with alpha channels better --- api/onnx_web/image/noise_source.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/image/noise_source.py b/api/onnx_web/image/noise_source.py index c543aa6e..a1dd47f1 100644 --- 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