Update handler.py

handler.py

@@ -22,10 +22,11 @@ class EndpointHandler():
 
         # pass inputs with all kwargs in data
         if parameters is not None:
-            embeddings = self.model.encode(inputs, **parameters)
+            embeddings = self.model.encode(inputs, **parameters)['dense_vecs']
         else:
-            embeddings = self.model.encode(inputs)
+            embeddings = self.model.encode(inputs)['dense_vecs']
         # postprocess the prediction
-        return embeddings
+        list_of_lists = [arr.tolist() for arr in embeddings]
+        return json.dumps(list_of_lists)
 
         # return self.model.encode(inputs, batch_size=12, max_length=8192)['dense_vecs']