start_text = "Polish my unrevealed"
encoded = tokenizer.encode(start_text)

# Add the batch dimension
encoded_tensor = torch.tensor(encoded).unsqueeze(0)

# Put the model into evaluation mode
model.eval()

out = generate_text_simple(
    model=model,
    idx=encoded_tensor,
    max_new_tokens=6,
    context_size=GPT_CONFIG_124M["context_length"]
)

decoded_text = tokenizer.decode(out.squeeze(0).tolist())
print(decoded_text)

        __
        __