UserWarning: X does not have valid feature names, but KNeighborsClassifier was fitted with feature names

This is caused since the training process uses a dataframe for the features while during inference time, a numpy array is passed. Can be fixed by changing the training script to use dataframe values, without column names.