Spaces:
Configuration error
Configuration error
oremaz
commited on
Commit
·
c5a7c47
1
Parent(s):
d1fc4da
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -265,6 +265,8 @@ class DynamicQueryEngineManager:
|
|
| 265 |
)
|
| 266 |
self.visual_reranker = ColPaliRerank(
|
| 267 |
top_n=3,
|
|
|
|
|
|
|
| 268 |
device="cpu"
|
| 269 |
)
|
| 270 |
def postprocess_nodes(self, nodes, query_bundle):
|
|
|
|
| 265 |
)
|
| 266 |
self.visual_reranker = ColPaliRerank(
|
| 267 |
top_n=3,
|
| 268 |
+
model="vidore/colpali-v1.2",
|
| 269 |
+
keep_retrieval_score=True,
|
| 270 |
device="cpu"
|
| 271 |
)
|
| 272 |
def postprocess_nodes(self, nodes, query_bundle):
|