Implementing a pretrained model