Input Image has to be greyscale, otherwise broadcast error. Thus add .convert('L') ab27eac hugruby commited on Feb 6, 2022