Below is the example that I am using, not sure if the libraries used here needs update for mlflow. 
Similarly for libraries used in config as well may need update accordingly.

genai-cookbook/rag_app_sample_code/A_POC_app/pdf_uc_volume/03_deploy_poc_to_review_app.py at main · ...