test / app.py
reshav1's picture
Update app.py
7bcce48 verified
raw
history blame
190 Bytes
import transformers
from transformers import AutoModelForCausalLM
model_id = "gpt2"
model = GPT2Model.from_pretrained('gpt2')
text = "Hello"
outputs = model.generate(text)
print(outputs)