Hi @Koushik Deb​  try this your problem will be resolve now. if it works don't forget to accept and upvote the answer 🙂

import mlflow
 
import pandas as pd
 
 
 
# Load model as a PyFuncModel.
 
logged_model = 'runs:/id/model'
 
loaded_model = mlflow.pyfunc.load_model(logged_model)
 
 
 
# Create a dataframe with a single column
 
data = pd.DataFrame({'text': ["visit www.bet365.com for a free trial"]})
 
 
 
# Call predict method with the dataframe
 
predictions = loaded_model.predict(data)