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