Spaces:
Sleeping
Sleeping
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 |
|