bart / app.py
xnetba's picture
Create app.py
c268f0b
raw
history blame
202 Bytes
from transformers import pipeline
generator = pipeline("text-generation", model="distilgpt2")
generator(
"In this course, we will teach you how to",
max_length=30,
num_return_sequences=2,
)