Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
change ksize in RAG
Browse files- RAG/colpali.py +1 -1
RAG/colpali.py
CHANGED
|
@@ -316,7 +316,7 @@ def colpali_search_rerank(query):
|
|
| 316 |
}
|
| 317 |
response = aos_client.search(
|
| 318 |
body = query_,
|
| 319 |
-
index = 'colpali-vs'
|
| 320 |
)
|
| 321 |
#print(response)
|
| 322 |
query_token_vectors = batch_embeddings
|
|
|
|
| 316 |
}
|
| 317 |
response = aos_client.search(
|
| 318 |
body = query_,
|
| 319 |
+
index = 'colpali-vs-reindex'
|
| 320 |
)
|
| 321 |
#print(response)
|
| 322 |
query_token_vectors = batch_embeddings
|