test / app.py
reshav1's picture
Update app.py
9b1d780 verified
raw
history blame
180 Bytes
import transformers
from transformers import GPT2LMHeadModel
model_id = "gpt2"
model = GPT2LMHeadModel.from_pretrained(model_id)
text = "Hello"
outputs = model(**encoded_text)