Spaces:
Sleeping
Sleeping
NCTCMumbai
commited on
Update app.py
Browse files
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(
|
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 |
|