deepwords / app.py
yash161101's picture
Update app.py
855a10b
raw
history blame
298 Bytes
from transformers import AutoModel, AutoTokenizer
model_name = "ml6team/gpt-2-small-conditional-quote-generator"
model = AutoModel.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
inputs = tokenizer("Hello world!", return_tensors="pt")
outputs = model(**inputs)