Charles Chan commited on
Commit
e85774e
·
1 Parent(s): 9aa294c
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -95,7 +95,7 @@ def answer_question(repo_id, temperature, max_length, question):
95
  with st.spinner("正在筛选本地数据集..."):
96
  question_embedding = st.session_state.embeddings.embed_query(question)
97
  # question_embedding_str = " ".join(map(str, question_embedding))
98
- docs_and_scores = st.session_state.db.similarity_search_by_vector(question_embedding, 8)
99
 
100
  context_list = []
101
  for doc, score in docs_and_scores:
 
95
  with st.spinner("正在筛选本地数据集..."):
96
  question_embedding = st.session_state.embeddings.embed_query(question)
97
  # question_embedding_str = " ".join(map(str, question_embedding))
98
+ docs_and_scores = st.session_state.db.similarity_search_by_vector(question_embedding)
99
 
100
  context_list = []
101
  for doc, score in docs_and_scores: