Update app.py

app.py

@@ -25,7 +25,7 @@ def greet_json():
     return {"Hello": "World!"}
 
 @app.post("/predict_list")
-async def predict(inp : InputListModel):
+async def predict_list(inp : InputListModel):
     sentence_pairs = [[query, doc] for query,doc in zip(inp.keywords, inp.contents)]
     scores = model.predict(sentence_pairs, convert_to_tensor=False)#.tolist()
     # (-scores).argsort().tolist()
@@ -37,7 +37,7 @@ async def predict(inp : InputModel):
     sentence_pairs = [[inp.keyword, inp.content]]
     scores = model.predict(sentence_pairs, convert_to_tensor=False)#.tolist()
     # (-scores).argsort().tolist()
-    return {"results":scores.tolist()}
+    return {"results":scores.tolist()[0]}
     
     # keywords = model.encode(inp.keywords)
     # contents = model.encode(inp.contents)