Update extract_abs.py

extract_abs.py

@@ -94,7 +94,7 @@ def get_diseases(sentence:str, GARD_dict:Dict[str,str], max_length:int) -> Tuple
 ## Section: Prepare ML/DL Models
 # This fuction prepares the model. Should call before running in notebook. -- The [Any] Type is a Huggingface Pipeline variable
 # Default with typing from here: https://stackoverflow.com/questions/38727520/how-do-i-add-default-parameters-to-functions-when-using-type-hinting
-def init_NER_pipeline(name_or_path_to_model_folder:str = "ncats/EpiExtract4GARD-v2") -> Tuple[Any, Set[str]]: #NER_pipeline, entities = init_NER_pipeline()
+def init_NER_pipeline(name_or_path_to_model_folder:str = "./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='simple')