radu.mutilica
commited on
Commit
·
d5609b4
1
Parent(s):
0de9240
logging and pydantic issues
Browse files- handler.py +3 -2
handler.py
CHANGED
@@ -48,15 +48,16 @@ class EndpointHandler:
|
|
48 |
f'Loaded {path}:top_k={self.top_k}:length={self.max_length}:workers={self.max_workers}')
|
49 |
|
50 |
def __call__(self, data: Dict[str, str]) -> List[Rank]:
|
51 |
-
inputs = data.pop('inputs')
|
52 |
-
|
53 |
logger.info(f'Received new docs to rerank: {data}')
|
|
|
|
|
54 |
ranks = self.model.rank(
|
55 |
inputs['query'],
|
56 |
inputs['documents'],
|
57 |
top_k=self.top_k,
|
58 |
num_workers=self.max_workers
|
59 |
)
|
|
|
60 |
|
61 |
ranks = [Rank(
|
62 |
corpus_id=r['corpus_id'],
|
|
|
48 |
f'Loaded {path}:top_k={self.top_k}:length={self.max_length}:workers={self.max_workers}')
|
49 |
|
50 |
def __call__(self, data: Dict[str, str]) -> List[Rank]:
|
|
|
|
|
51 |
logger.info(f'Received new docs to rerank: {data}')
|
52 |
+
|
53 |
+
inputs = data.pop('inputs')
|
54 |
ranks = self.model.rank(
|
55 |
inputs['query'],
|
56 |
inputs['documents'],
|
57 |
top_k=self.top_k,
|
58 |
num_workers=self.max_workers
|
59 |
)
|
60 |
+
logger.info(f'New ranks: {ranks}')
|
61 |
|
62 |
ranks = [Rank(
|
63 |
corpus_id=r['corpus_id'],
|