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