dnlp-demo / main.py
Jackss
Added app file
ac5a31f
raw
history blame
242 Bytes
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"]}