gpt2-space / app.py
SunDay-s's picture
Update app.py
6f90ddb verified
raw
history blame contribute delete
271 Bytes
from transformers import GPT2Tokenizer, GPT2Model
tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
model = GPT2Model.from_pretrained('gpt2')
text = "Replace me by any text you'd like."
encoded_input = tokenizer(text, return_tensors='pt')
output = model(**encoded_input)