added model

handler.py

@@ -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']
+        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))\