Ah, the model expects 3 channels in the input. Since, I believe the model is already trained, you can concatenate the image 3 times by adding this
image = np.concatenate([image, image, image], 0)

after
image = image.transpose(2, 0, 1)