milestone-2 / app.py
cdb24's picture
Update app.py
7cc7bb3
raw
history blame
165 Bytes
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("gpt2")
model = AutoModelForCausalLM.from_pretrained("gpt2")