Making predictions using the trained model