Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def greet_json():
|
|
20 |
return {"Hello": "World!"}
|
21 |
|
22 |
@app.post("/predict")
|
23 |
-
def predict(inp : InputModel):
|
24 |
sentence_pairs = [[query, doc] for query,doc in zip(inp.keywords, inp.contents)]
|
25 |
scores = model.predict(sentence_pairs, convert_to_tensor=False)#.tolist()
|
26 |
# (-scores).argsort().tolist()
|
|
|
20 |
return {"Hello": "World!"}
|
21 |
|
22 |
@app.post("/predict")
|
23 |
+
async def predict(inp : InputModel):
|
24 |
sentence_pairs = [[query, doc] for query,doc in zip(inp.keywords, inp.contents)]
|
25 |
scores = model.predict(sentence_pairs, convert_to_tensor=False)#.tolist()
|
26 |
# (-scores).argsort().tolist()
|