How to test grayscale images?

Hello! Thanks for your codes! But I have a problem about testing grayscale images. Could you please help me with that problem? Thank you very much!

IndexError: index 1 is out of bounds for dimension 0 with size 1