|
|
|
@ -67,7 +67,7 @@ def update_dataset():
|
|
|
|
|
# 获取匹配的接口
|
|
|
|
|
@app.route('/matches', methods=['POST'])
|
|
|
|
|
def get_matches():
|
|
|
|
|
query_sentence = request.json.get('query_sentence', '')
|
|
|
|
|
query_sentence = request.json.get('querySentence', '')
|
|
|
|
|
query_embedding = model.encode([query_sentence])[0]
|
|
|
|
|
# 对向量进行单位化
|
|
|
|
|
query_embedding = query_embedding / np.linalg.norm(query_embedding)
|
|
|
|
@ -85,7 +85,7 @@ def get_matches():
|
|
|
|
|
# 获取所有相似度的接口
|
|
|
|
|
@app.route('/get_all_similarities', methods=['POST'])
|
|
|
|
|
def get_all_similarities():
|
|
|
|
|
query_sentence = request.json.get('query_sentence', '')
|
|
|
|
|
query_sentence = request.json.get('querySentence', '')
|
|
|
|
|
query_embedding = model.encode([query_sentence])[0]
|
|
|
|
|
# 对向量进行单位化
|
|
|
|
|
query_embedding = query_embedding / np.linalg.norm(query_embedding)
|
|
|
|
@ -98,4 +98,4 @@ def get_all_similarities():
|
|
|
|
|
return jsonify({'status': 'success', 'results': results})
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
app.run(debug=True, host='0.0.0.0')
|
|
|
|
|
app.run(debug=True, host='0.0.0.0')
|