rasa : 修改 querySentence

dev_2.1.0
xueqingkun 1 year ago
parent 597c865d84
commit 2f522b03a2

@ -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')
Loading…
Cancel
Save