save

src/streamlit_app.py

@@ -61,6 +61,10 @@ uniques_filename = os.path.join(output_dir, "uniques_data.pickle")
 with open(uniques_filename, "wb") as output_pickle:
     pickle.dump(uniques, output_pickle)
 
+
+st.write("Model saved to {}".format(model_filename)
+st.writ("Click below to download the model."))
+
 # Load the files to enable download
 with open(model_filename, "rb") as f:
     model_bytes = f.read()