@Harvinder Singh I’d recommend the approach @Omri Har-Shemesh shared above. You can also build a second pipeline that pulls trained model from MLFlow and then package it with BentoML for serving. Although managing two separate pipeline takes more time and could introduce more issues.