classify now returns predictions from the model