ML Model Integration