Update handler.py

handler.py

@@ -53,9 +53,9 @@ class EndpointHandler():
             print("E")
 
             # Perform pooling. In this case, max pooling.    
-            embedding = mean_pooling(model_output, encoded_input['attention_mask'])
+            # embedding = mean_pooling(model_output, encoded_input['attention_mask'])
             print("F")
             
-            sentence_embeddings.append(embedding.tolist())
+            sentence_embeddings.append(model_output)
             print("G")
         return sentence_embeddings