from transformers import pipeline

def load_model():
    pipe = pipeline("text-generation", model="distilgpt2")
    return pipe