4b857ac
1
2
3
4
5
6
from transformers import pipeline def load_model(): pipe = pipeline("text-generation", model="distilgpt2") return pipe
