File size: 242 Bytes
d1659f7
 
 
 
ac5a31f
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from fastapi import FastAPI

app = FastAPI()

# @app.get('/')
# def read_root():
#     return {"Hello": "World!"}

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