Update handler.py

handler.py

@@ -37,14 +37,23 @@ class EndpointHandler():
       Return:
             A :obj:`list` | `dict`: will be serialized and returned
         """
+        print("A")
         sentences = data.pop("inputs",data)
+        print("B")
         sentence_embeddings = []
+        print("C")
         for sentence in sentences:    
             encoded_input = self.tokenizer(sentence, padding=True, truncation=True, return_tensors='pt')    
+            print("D")
+
             # Compute token embeddings    
             with torch.no_grad():       
-                model_output = self.onnx_extractor(**encoded_input)    
+                model_output = self.onnx_extractor(**encoded_input)  
+            print("E")
+
             # Perform pooling. In this case, max pooling.    
             embedding = mean_pooling(model_output, encoded_input['attention_mask'])
+            print("F")
             sentence_embeddings.append(embedding.tolist())
+            print("G")
         return sentence_embeddings