Update deploy_pipeline.py

deploy_pipeline.py

@@ -74,17 +74,18 @@ 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, tokenizer_results['tokens']
+        return {'outputs': outputs, 'tokens': tokenizer_results['tokens']}
+
+    def postprocess(self, outputs):
+        
+        model_outputs = outputs['outputs']
+        tokens = outputs['tokens']
 
-    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]
-        
-        print(tokenizer_results)
-        
-        return model_outputs[0], tokenizer_results
-
+                
+        return model_outputs[0], tokens
 
 def main():