Using the pretrained model for prediction