Spaces:
Configuration error
Configuration error
oremaz
commited on
Commit
·
bd4ad64
1
Parent(s):
999040b
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -396,7 +396,8 @@ class DynamicQueryEngineManager:
|
|
| 396 |
def __init__(self):
|
| 397 |
self.text_reranker = SentenceTransformerRerank(
|
| 398 |
model="cross-encoder/ms-marco-MiniLM-L-2-v2",
|
| 399 |
-
top_n=3
|
|
|
|
| 400 |
)
|
| 401 |
|
| 402 |
self.visual_reranker = ColPaliRerank(
|
|
|
|
| 396 |
def __init__(self):
|
| 397 |
self.text_reranker = SentenceTransformerRerank(
|
| 398 |
model="cross-encoder/ms-marco-MiniLM-L-2-v2",
|
| 399 |
+
top_n=3,
|
| 400 |
+
device = "cpu"
|
| 401 |
)
|
| 402 |
|
| 403 |
self.visual_reranker = ColPaliRerank(
|