dsv

src/streamlit_app.py

@@ -60,3 +60,13 @@ with open(model_filename, "wb") as rf_pickle:
 uniques_filename = os.path.join(output_dir, "uniques_data.pickle")
 with open(uniques_filename, "wb") as output_pickle:
     pickle.dump(uniques, output_pickle)
+
+# Load the files to enable download
+with open(model_filename, "rb") as f:
+    model_bytes = f.read()
+st.download_button(
+    label="Download Trained Model (random_forest_penguin.pickle)",
+    data=model_bytes,
+    file_name="random_forest_penguin.pickle",
+    mime="application/octet-stream"
+)