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"]}