Update deploy_pipeline.py

deploy_pipeline.py

@@ -41,7 +41,6 @@ class TokenizeAndAlignLabelsStep():
         return tokenized_inputs
 
 
-
 class BERT_CRF_Pipeline(Pipeline):
 
     def _sanitize_parameters(self, **kwargs):
@@ -75,14 +74,16 @@ class BERT_CRF_Pipeline(Pipeline):
         outputs = self.model(input_ids=input_ids, token_type_ids=token_type_ids,
                              attention_mask=attention_mask, labels=None, labels_mask=labels_mask)
 
-        return outputs
+        return outputs, tokenizer_results['tokens']
 
-    def postprocess(self, model_outputs):
+    def postprocess(self, model_outputs, tokenizer_results):
         # From Ner_tags to Ner_labels
         for i, label in enumerate(model_outputs[0]):
             model_outputs[0][i] = self.model.config.id2label[label]
-
-        return model_outputs[0]
+        
+        print(tokenizer_results)
+        
+        return model_outputs[0], tokenizer_results
 
 
 def main():