Update app.py

app.py

@@ -23,6 +23,13 @@ st.markdown('##### The output contains smiles of predicted products and sum of l
 
 display_text = 'input the reaction smiles (e.g. REACTANT:CNc1nc(SC)ncc1CO.O.O=[Cr](=O)([O-])O[Cr](=O)(=O)[O-].[Na+]CATALYST: REAGENT: SOLVENT:CC(=O)O)'
 
+st.download_button(
+                label="Download demo_input.csv",
+                data=pd.read_csv('demo_input.csv'),
+                file_name='demo_input.csv',
+                mime='text/csv',
+            )
+
 class CFG():
     num_beams = st.number_input(label='num beams', min_value=1, max_value=10, value=5, step=1)
     num_return_sequences = num_beams