Update app.py

app.py

@@ -63,10 +63,10 @@ def load_models():
     return classify_tokenizer, classify_model, NER_pipeline, entity_classes, GARD_dict, max_length
 
 @st.cache
- def convert_df(df):
-     # IMPORTANT: Cache the conversion to prevent computation on every rerun
-     return df.to_csv().encode('utf-8')
-
+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):
@@ -111,9 +111,10 @@ if disease_or_gard_id:
                                 extract_diseases,GARD_dict, max_length, 
                                 classify_model_vars)
     st.dataframe(df, height=100)
+    csv = convert_df(df)
     st.download_button(
         label="Download epidemiology results for "+disease_or_gard_id+" as CSV",
-        data=df.to_csv().encode('utf-8'),
+        data = csv,
         file_name=disease_or_gard_id+'.csv',
         mime='text/csv',
         )