wiki / app.py
faztrick's picture
Create app.py
8e67708
raw
history blame contribute delete
No virus
165 Bytes
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("gpt2")
model = AutoModelForCausalLM.from_pretrained("gpt2")