nicolaebanari commited on
Commit
c6afee5
·
verified ·
1 Parent(s): 4d40ef2

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +5 -2
handler.py CHANGED
@@ -15,8 +15,11 @@ class EndpointHandler:
15
  logger.info("Inference started")
16
  logger.info(data)
17
  logger.info(type(data))
18
- score = self.reranker.compute_score([data['query'], data['text']])
 
 
 
19
 
20
- output = {"score": score}
21
  return output
22
 
 
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}
24
  return output
25