Update extract_abs.py

extract_abs.py

@@ -97,7 +97,7 @@ def get_diseases(sentence:str, GARD_dict:Dict[str,str], max_length:int) -> Tuple
 def init_NER_pipeline(name_or_path_to_model_folder:str = "ncats/EpiExtract4GARD-v2") -> Tuple[Any, Set[str]]: #NER_pipeline, entities = init_NER_pipeline()
     tokenizer = BertTokenizer.from_pretrained(name_or_path_to_model_folder)
     custommodel = AutoModelForTokenClassification.from_pretrained(name_or_path_to_model_folder)
-    customNER = pipeline('ner', custommodel, tokenizer=tokenizer, aggregation_strategy='none')
+    customNER = pipeline('ner', custommodel, tokenizer=tokenizer, aggregation_strategy='simple')
     
     config = BertConfig.from_pretrained(name_or_path_to_model_folder)
     labels = {re.sub(".-","",label) for label in config.label2id.keys() if label != "O"}