File size: 180 Bytes
360e70e
9bcb61e
139bee4
360e70e
7bcce48
360e70e
 
0c33b7e
 
360e70e
1
2
3
4
5
6
7
8
9
10
import transformers 
from transformers import AGPT2Model
model_id = "gpt2"

model = GPT2Model.from_pretrained('gpt2')

text = "Hello"

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