wzkariampuzha commited on
Commit
4e45f70
1 Parent(s): 80b9099

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,10 +51,10 @@ def load_models():
51
  #classify_model_vars = classify_abs.init_classify_model()
52
  NER_pipeline, entity_classes = extract_abs.init_NER_pipeline()
53
  GARD_dict, max_length = extract_abs.load_GARD_diseases()
54
- return nlpSci, nlpSci2, classify_tokenizer, classify_model, NER_pipeline, entity_classes, GARD_dict, max_length
55
 
56
  with st.spinner('Loading Epidemiology Models and Dependencies...'):
57
- nlpSci, nlpSci2, classify_tokenizer, classify_model, NER_pipeline, entity_classes, GARD_dict, max_length = load_models()
58
  #Load spaCy models which cannot be cached due to hash function error
59
  nlp = spacy.load('en_core_web_lg')
60
  nlpSci = spacy.load("en_ner_bc5cdr_md")
 
51
  #classify_model_vars = classify_abs.init_classify_model()
52
  NER_pipeline, entity_classes = extract_abs.init_NER_pipeline()
53
  GARD_dict, max_length = extract_abs.load_GARD_diseases()
54
+ return classify_tokenizer, classify_model, NER_pipeline, entity_classes, GARD_dict, max_length
55
 
56
  with st.spinner('Loading Epidemiology Models and Dependencies...'):
57
+ classify_tokenizer, classify_model, NER_pipeline, entity_classes, GARD_dict, max_length = load_models()
58
  #Load spaCy models which cannot be cached due to hash function error
59
  nlp = spacy.load('en_core_web_lg')
60
  nlpSci = spacy.load("en_ner_bc5cdr_md")