Update handler.py

handler.py

@@ -29,4 +29,4 @@ class EndpointHandler():
 
         # Perform pooling. In this case, max pooling.
         sentence_embeddings = mean_pooling(model_output, encoded_input['attention_mask'])
-        return model_output.tolist()
+        return sentence_embeddings.tolist()