python import torch from transformers import GPT2LMHeadModel, GPT2Tokenizer from contexts import ContextModel, ContextGenerator model_name = "gpt2" model = GPT2LMHeadModel.from_pretrained(model_name) tokenizer = GPT2Tokenizer.from_pretrained(model_name) context_generator = ContextGenerator(model, tokenizer) generated_text = context_generator.generate_text(context, max_length=100)