Get startedGet started for free

Text completion with pre-trained GPT-2 models

Back at PyBooks, your current project involves creating captivating narratives based on existing stories to engage customers and enhance their experience. To achieve this, you need a powerful text generation tool that can seamlessly generate compelling text continuations. You'll be using a pre-trained model to get the job done.

The following has been loaded for you: torch and GPT2Tokenizer,GPT2LMHeadModel from transformers.

This exercise is part of the course

Deep Learning for Text with PyTorch

View Course

Hands-on interactive exercise

Have a go at this exercise by completing this sample code.

# Initialize the tokenizer
tokenizer = ____

# Initialize the pre-trained model
model = ____
Edit and Run Code