File size: 170 Bytes
360e70e
04020fa
139bee4
360e70e
04020fa
360e70e
 
0c33b7e
68ac2cc
1
2
3
4
5
6
7
8
9
10
import transformers 
from transformers import GPT2LMHeadModel
model_id = "gpt2"

model = GPT2LMHeadModel.from_pretrained(model_id)

text = "Hello"

outputs = model(text)