Update handler.py
Browse files- handler.py +5 -1
handler.py
CHANGED
|
@@ -394,7 +394,11 @@ class EndpointHandler:
|
|
| 394 |
with open(rerank_name_path, "r", encoding="utf-8") as f:
|
| 395 |
rerank_name = f.read().strip()
|
| 396 |
if rerank_name:
|
| 397 |
-
self.reranker = CrossEncoder(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 398 |
|
| 399 |
# -----------------------------
|
| 400 |
# Input normalization / query formatting
|
|
|
|
| 394 |
with open(rerank_name_path, "r", encoding="utf-8") as f:
|
| 395 |
rerank_name = f.read().strip()
|
| 396 |
if rerank_name:
|
| 397 |
+
self.reranker = CrossEncoder(
|
| 398 |
+
rerank_name,
|
| 399 |
+
max_length=512,
|
| 400 |
+
device=self.device,
|
| 401 |
+
)
|
| 402 |
|
| 403 |
# -----------------------------
|
| 404 |
# Input normalization / query formatting
|