Narya-ai
/

lr-bert-base-uncased

Inference Endpoints

Model card Files Files and versions Community

Uan Sholanbayev commited on Oct 13, 2023

Commit

582955e

•

1 Parent(s): 1d62fed

added model

Files changed (1) hide show

handler.py +3 -2

handler.py CHANGED Viewed

@@ -28,8 +28,9 @@ class EndpointHandler():
         return outputs.last_hidden_state.mean(dim=1).cpu().numpy()
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        queries = data['queries']
-        texts = data['texts']
         queries_vec = self.get_embeddings(queries)
         texts_vec = self.get_embeddings(texts)
         diff = (np.array(texts_vec)[:, np.newaxis] - np.array(queries_vec))\

         return outputs.last_hidden_state.mean(dim=1).cpu().numpy()
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+        inputs = data.pop("inputs",data)
+        queries = inputs['queries']
+        texts = inputs['texts']
         queries_vec = self.get_embeddings(queries)
         texts_vec = self.get_embeddings(texts)
         diff = (np.array(texts_vec)[:, np.newaxis] - np.array(queries_vec))\