How to get predictions from an ML model