from transformers import pipeline

generator = pipeline(task="text-generation")