File size: 597 Bytes
a52f35b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from fastapi import FastAPI
from transformers import pipeline


#create a new FASTAPI App instances 
app=FastAPI()


##create a new pipeline instances
pipe = pipeline("text2text-generation", model="google/flan-t5-small")

@app.get("/")
def home():
    return{"message":"Hello World !"}


#Define a function o handle the GET request at '/generate'


@app.get("/generate")
def generate(text:str):
    ###use the pipeline to generate text from given input text

    output=pipe(text)


    ##return the generate text in Json response

    return{"output":output[0][' ']}