Spaces:
Configuration error
Configuration error
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(
|