@Dan Zafar I mentioned the incorrect path in the original question but I did train the model with correct file.

!wget https://raw.githubusercontent.com/mlflow/mlflow-example/master/wine-quality.csv

There is no issues when trying to predict using the h2o model object. But the prediction fails when using the MLFLOW's pyfunc flavour