File size: 268 Bytes
d1659f7 12bb371 d1659f7 b75d406 ac5a31f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from fastapi import FastAPI
model_name = 'allenai/specter'
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"]} |