File size: 173 Bytes
134505f
 
 
 
 
 
 
10631e7
1
2
3
4
5
6
7
8
9
from transformers import pipeline

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


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