victor7246 commited on
Commit
7d4c84c
·
verified ·
1 Parent(s): e789ac8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ if __name__ == '__main__':
71
  q_type = extract_question_type(llm, question)
72
  if q_type == 'specific':
73
  with st.status("Retrieving results..."):
74
- top_table_names = table_search(question, topk=3)['table'].tolist()
75
  print (top_table_names)
76
  try:
77
  db_chain._call(inputs={'query': question, 'history': [], \
 
71
  q_type = extract_question_type(llm, question)
72
  if q_type == 'specific':
73
  with st.status("Retrieving results..."):
74
+ top_table_names = table_search(question, topk=1)['table'].tolist()
75
  print (top_table_names)
76
  try:
77
  db_chain._call(inputs={'query': question, 'history': [], \