wzkariampuzha commited on
Commit
7a38dc3
1 Parent(s): a38aa29

Update classify_abs.py

Browse files
Files changed (1) hide show
  1. classify_abs.py +1 -1
classify_abs.py CHANGED
@@ -58,7 +58,7 @@ def standardizeSciTerms(abstract:str, nlpSci:Any, nlpSci2:Any) -> str:
58
 
59
  # Prepare model
60
  #nlp, nlpSci, nlpSci2, classify_model, classify_tokenizer= init_classify_model()
61
- def init_classify_model(model:str='my_model_orphanet_final') -> Tuple[Any,Any,Any,Any,Any]:
62
  #Load spaCy models
63
  nlp = spacy.load('en_core_web_lg')
64
  nlpSci = spacy.load("en_ner_bc5cdr_md")
 
58
 
59
  # Prepare model
60
  #nlp, nlpSci, nlpSci2, classify_model, classify_tokenizer= init_classify_model()
61
+ def init_classify_model(model:str='LSTM_RNN_Model') -> Tuple[Any,Any,Any,Any,Any]:
62
  #Load spaCy models
63
  nlp = spacy.load('en_core_web_lg')
64
  nlpSci = spacy.load("en_ner_bc5cdr_md")