mlflow is throwing error for the shape of input

Question

I am running the code for prediction which will take the model from mlflow deployment. Code I have copied from the example given by mlflow

import mlflow
logged_model = 'runs:/id/model'

# Load model as a PyFuncModel.
loaded_model = mlflow.pyfunc.load_model(logged_model)

# Predict on a Pandas DataFrame.
import pandas as pd
loaded_model.predict(pd.DataFrame(data))

Here I am getting the 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, "Some big integer").