After executing the given code by you, I have got this error

MlflowException: The input pandas dataframe column 'None' contains scalar values, which requires the shape to be (-1,), but got tensor spec shape of (-1, 14654).

Let me know if you have any idea on this.