how to convert the following model from RGB into 1 chanel (gray only)

Question

I need to change the following code into 1 color (gray) instead of RGB, so how can I do it ?

#model training and saving best model best_accuracy = 0.0 for epoch in range(