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