Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -27,7 +27,7 @@ class EndpointHandler:
|
|
27 |
|
28 |
# Extract inputs and type
|
29 |
texts = data.pop("inputs")
|
30 |
-
return sbert.encode(
|
31 |
texts,
|
32 |
task="retrieval.query"
|
33 |
).tolist()
|
|
|
27 |
|
28 |
# Extract inputs and type
|
29 |
texts = data.pop("inputs")
|
30 |
+
return self.sbert.encode(
|
31 |
texts,
|
32 |
task="retrieval.query"
|
33 |
).tolist()
|