model_loader.py

from transformers import pipeline

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