Update app.py

app.py

@@ -67,12 +67,12 @@ def bot(history, api_kind):
     print(query)
     query_vec_flat = [arr.flatten() for arr in query_vec]
     logger.warning(f'Finished query vec')
-    documents = table.search(query_vec_flat, vector_column_name=VECTOR_COLUMN_NAME).limit(top_k_rank).to_list()
+    #documents = table.search(query_vec_flat, vector_column_name=VECTOR_COLUMN_NAME).limit(top_k_rank).to_list()
 
 
 
     logger.warning(f'Finished search')
-    #documents = table.search(query_vec, vector_column_name=VECTOR_COLUMN_NAME).limit(top_rerank).to_list()
+    documents = table.search(query_vec, vector_column_name=VECTOR_COLUMN_NAME).limit(top_rerank).to_list()
     documents = [doc[TEXT_COLUMN_NAME] for doc in documents]
     logger.warning(f'start cross encoder {len(documents)}')
     # Retrieve documents relevant to query