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