test-B / t5.py
projects-jeru's picture
Duplicate from projects-jeru/container-test
134505f
raw
history blame
152 Bytes
from transformers import pipeline
pipe_flan = pipeline(model="bigscience/bloom-560m")
def t5(input):
output = pipe_flan(input)
return output