File size: 228 Bytes
e799e02
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
from transformers import pipeline

pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")

@app.get("/infer_t5")
def t5(input):
    output = pipe_flan(input)
    return {"output": output[0]["generated_text"]}