Spaces:
Sleeping
Sleeping
wzkariampuzha
commited on
Commit
•
4e45f70
1
Parent(s):
80b9099
Update app.py
Browse files
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
|
55 |
|
56 |
with st.spinner('Loading Epidemiology Models and Dependencies...'):
|
57 |
-
|
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")
|