louiecerv commited on
Commit
567000e
·
1 Parent(s): e200808
Files changed (1) hide show
  1. src/streamlit_app.py +10 -0
src/streamlit_app.py CHANGED
@@ -73,4 +73,14 @@ st.download_button(
73
  data=model_bytes,
74
  file_name="random_forest_penguin.pickle",
75
  mime="application/octet-stream"
 
 
 
 
 
 
 
 
 
 
76
  )
 
73
  data=model_bytes,
74
  file_name="random_forest_penguin.pickle",
75
  mime="application/octet-stream"
76
+ )
77
+
78
+ # Load the files to enable download
79
+ with open(unique_filename, "rb") as f:
80
+ model_bytes = f.read()
81
+ st.download_button(
82
+ label="Download the data (uniques_data.pickle)",
83
+ data=model_bytes,
84
+ file_name="runiques_data.pickle",
85
+ mime="application/octet-stream"
86
  )