Amer-Alhamvi

is the image B&W image? If so I think this may be the problem try converting it into first rgb and then passing it to the model