Update handler.py
Browse files- 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
|
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']
|