Update app.py

app.py

@@ -61,6 +61,7 @@ def load_models():
     GARD_dict, max_length = extract_abs.load_GARD_diseases()
     return classify_tokenizer, classify_model, NER_pipeline, entity_classes, GARD_dict, max_length
 
+@st.experimental_memo
 def epi_sankey(sankey_data):
     gathered, relevant, epidemiologic = sankey_data