Import the pretrained TensorFlow model