NCTCMumbai commited on
Commit
54d140a
·
verified ·
1 Parent(s): 30c25e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -63,8 +63,9 @@ def bot(history, api_kind):
63
  document_start = perf_counter()
64
 
65
  query_vec = retriever.encode(query)
 
66
  logger.warning(f'Finished query vec')
67
- doc1 = table.search(query_vec, vector_column_name=VECTOR_COLUMN_NAME).limit(top_k_rank)
68
 
69
 
70
 
 
63
  document_start = perf_counter()
64
 
65
  query_vec = retriever.encode(query)
66
+ query_vec_flat = [arr.flatten() for arr in query_vec]
67
  logger.warning(f'Finished query vec')
68
+ doc1 = table.search(query_vec_flat, vector_column_name=VECTOR_COLUMN_NAME).limit(top_k_rank)
69
 
70
 
71