Isolate model predictions functionality