Spaces:
Sleeping
Sleeping
File size: 287 Bytes
e7799a9 578e1ad d7e046c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
model_id = "gpt2"
model = GPT2LMHeadModel.from_pretrained(model_id)
tokenizer = GPT2Tokenizer.from_pretrained(model_id)
text = "Hello"
encoded_text = tokenizer(text, return_tensors="pt")
outputs = model(**encoded_text) |