t5.py

from transformers import pipeline

pipe_flan = pipeline(model="bigscience/bloom-560m")


def t5(input):
    output = pipe_flan(input)
    return output[0]["generated_text"]