zkwang commited on
Commit
f4c9378
1 Parent(s): 201c562

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -2
handler.py CHANGED
@@ -1,4 +1,3 @@
1
- import json
2
  from typing import Dict, List, Any
3
  from FlagEmbedding import BGEM3FlagModel
4
 
@@ -28,6 +27,6 @@ class EndpointHandler():
28
  embeddings = self.model.encode(inputs)['dense_vecs']
29
  # postprocess the prediction
30
  list_of_lists = [arr.tolist() for arr in embeddings]
31
- return json.dumps(list_of_lists)
32
 
33
  # return self.model.encode(inputs, batch_size=12, max_length=8192)['dense_vecs']
 
 
1
  from typing import Dict, List, Any
2
  from FlagEmbedding import BGEM3FlagModel
3
 
 
27
  embeddings = self.model.encode(inputs)['dense_vecs']
28
  # postprocess the prediction
29
  list_of_lists = [arr.tolist() for arr in embeddings]
30
+ return list_of_lists
31
 
32
  # return self.model.encode(inputs, batch_size=12, max_length=8192)['dense_vecs']