predict.py

from summarizer import Summarizer
from transformers import pipeline


def summarize(x):
    summarizer = Summarizer(model='distilbert-base-uncased')
    output = summarizer(x, ratio=0.2, return_as_list=False)
    return output