Using the trained model to predict