Update epi_pipeline.py

epi_pipeline.py

@@ -659,10 +659,13 @@ class NER_Pipeline:
     
     def __str__(self):
         return "Instantiation: pipe = NER_Pipeline(name_or_path_to_model_folder)"+"\n Calling: output_dict = pipe(text)"
-    
+
+    def __call__(self, text:str, rd_identify:Union[GARD_Search,None] = None):
+        return self.getTextExtraction(text, rd_identify)
+        
     #Custom pipeline by WKariampuzha @NCATS (not Huggingface/Google/NVIDIA copyright)
     @timeit
-    def __call__(self, text:str, rd_identify:Union[GARD_Search,None] = None):
+    def getTextExtraction(self, text:str, rd_identify:Union[GARD_Search,None] = None):
         output_dict = {label:[] for label in self.labels}
         
         dataset = NerDataset(text, self.bert_tokenizer, self.config)