My model is something like this and designed to work with RGB images. Previously I tested with both RGB and Grayscale images and it was fine. Suddenly it is not working with