Update app.py

app.py

@@ -62,7 +62,13 @@ 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
+@st.cache
+ def convert_df(df):
+     # IMPORTANT: Cache the conversion to prevent computation on every rerun
+     return df.to_csv().encode('utf-8')
+
+#@st.experimental_memo
+@st.cache(allow_output_mutation=True)
 def epi_sankey(sankey_data):
     gathered, relevant, epidemiologic = sankey_data