jpohhhh commited on
Commit
1d6ef8a
1 Parent(s): 2f0ed53

Revert "Update handler.py"

Browse files

This reverts commit 2f0ed534bf8dd02bc179f1a41916123d1997eb3c.

Files changed (1) hide show
  1. 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.last_hidden_state.tolist()
 
34
  # Perform pooling
35
  sentence_embeddings = mean_pooling(outputs, encoded_inputs['attention_mask'])
36
  # postprocess the prediction
37
+ return outputs.tolist()