Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
| 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': [], \
|