Predict colored version of gray image with using KNN in Machine Learning

Question

I have a source image with rgb color shape is (422, 750, 3) I created grayscale version of image with gray_image = imread(\'source.png\', as_gray=True). Then I extr