return embeddings list directly

handler.py

@@ -16,7 +16,7 @@ class EndpointHandler():
         self.tokenizer = AutoTokenizer.from_pretrained(path)
         self.model = AutoModel.from_pretrained(path)
 
-    def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
+    def __call__(self, data: Dict[str, Any]) -> List[List[int]]:
         """
        data args:
             inputs (:obj: `List[str]`)
@@ -32,4 +32,4 @@ class EndpointHandler():
 
         embeddings = average_pool(outputs.last_hidden_state, batch_dict['attention_mask']).tolist()
 
-        return {"embeddings": embeddings}
+        return embeddings