Update app.py

app.py

@@ -61,7 +61,11 @@ def load_models():
     NER_pipeline, entity_classes = extract_abs.init_NER_pipeline()
     GARD_dict, max_length = extract_abs.load_GARD_diseases()
     return classify_tokenizer, classify_model, NER_pipeline, entity_classes, GARD_dict, max_length
-    
+
+def convert_df(df):
+     # IMPORTANT: Cache the conversion to prevent computation on every rerun
+     return df.to_csv().encode('utf-8')
+
 with st.spinner('Loading Epidemiology Models and Dependencies...'):
     classify_model_vars, NER_pipeline, entity_classes, GARD_dict, max_length = load_models_experimental()
     #classify_tokenizer, classify_model, NER_pipeline, entity_classes, GARD_dict, max_length = load_models()
@@ -81,6 +85,13 @@ if disease_or_gard_id:
                                         extract_diseases,GARD_dict, max_length, 
                                         classify_model_vars)
   st.dataframe(df)
+  csv = convert_df(df)
+  st.download_button(
+     label="Download epidemiology results for "+disease_or_gard_id+" as CSV",
+     data=csv,
+     file_name=disease_or_gard_id+'.csv',
+     mime='text/csv',
+ )
   #st.dataframe(data=None, width=None, height=None)
   
 # st.code(body, language="python")