Customizing model serving output with MLflow PyFunc