import transformers 
from transformers import AGPT2Model
model_id = "gpt2"

model = GPT2Model.from_pretrained('gpt2')

text = "Hello"

outputs = model.generate(text)
print(outputs)