cli.py dies on greyscale-mode PNGs

[wareya]

add this or something equivalent to it before transposing the array:

if in_img.ndim == 2:
    in_img = np.expand_dims(in_img, axis=-1)