Update handler.py

handler.py

@@ -8,7 +8,7 @@ logger.info("INFO")
 
 class EndpointHandler:
     def __init__(self, path=""):
-        self.reranker = FlagReranker('BAAI/bge-m3', use_fp16=True) 
+        self.reranker = FlagReranker('BAAI/bge-reranker-large', use_fp16=True) 
 
     def __call__(self, inputs):
         data = inputs['inputs']
@@ -17,7 +17,7 @@ class EndpointHandler:
         logger.info(type(data))
         scores = []
         for t in data['texts']:
-            score = self.reranker.compute_score([[data['query'], t]])
+            score = self.reranker.compute_score([data['query'], t])
             logger.info(score)
             scores.append(score)