nicolaebanari commited on
Commit
9ea6a04
·
verified ·
1 Parent(s): e1bc062

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -1
handler.py CHANGED
@@ -13,11 +13,12 @@ class EndpointHandler:
13
  def __call__(self, inputs):
14
  data = inputs['inputs']
15
  logger.info("Inference started")
16
- logger.info(data)
17
  logger.info(type(data))
18
  scores = []
19
  for t in data['texts']:
20
  score = self.reranker.compute_score([[data['query'], t]])
 
21
  scores.append(score)
22
 
23
  output = {"scores": scores}
 
13
  def __call__(self, inputs):
14
  data = inputs['inputs']
15
  logger.info("Inference started")
16
+ # logger.info(data)
17
  logger.info(type(data))
18
  scores = []
19
  for t in data['texts']:
20
  score = self.reranker.compute_score([[data['query'], t]])
21
+ logger.info(score)
22
  scores.append(score)
23
 
24
  output = {"scores": scores}