test-project / app.py
RrDev's picture
Update app.py
fc084fd
raw
history blame contribute delete
221 Bytes
from transformers import pipeline
generator = pipeline("text-generation", model="distilgpt2")
res = generator(
"In this course we will teach you how to",
max_length = 30,
num_return_sequence = 2
)
print(res)