Spaces:
Sleeping
Sleeping
File size: 286 Bytes
aa97cc1 d7e046c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
rom 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) |