Ai-Guru / app.py
enzer1992's picture
Update app.py
40819c2 verified
raw
history blame
188 Bytes
from transformers import TFGPT2LMHeadModel, GPT2Tokenizer
model_name = "gpt2"
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
model = TFGPT2LMHeadModel.from_pretrained(model_name)