Revert "Update handler.py"
Browse filesThis reverts commit 2f0ed534bf8dd02bc179f1a41916123d1997eb3c.
- handler.py +1 -1
handler.py
CHANGED
@@ -34,4 +34,4 @@ class EndpointHandler():
|
|
34 |
# Perform pooling
|
35 |
sentence_embeddings = mean_pooling(outputs, encoded_inputs['attention_mask'])
|
36 |
# postprocess the prediction
|
37 |
-
return outputs.
|
|
|
34 |
# Perform pooling
|
35 |
sentence_embeddings = mean_pooling(outputs, encoded_inputs['attention_mask'])
|
36 |
# postprocess the prediction
|
37 |
+
return outputs.tolist()
|