import streamlit as st from tools import * _, file_tree_ph = sidebar(globals()) st.title("Download models") exp_id = st.session_state.get('exp_id') if exp_id is None: st.write("Please select an experiment first.") st.stop() with open(f'{PATH_TO_MODELS}/best_model.pkl', 'rb') as f: st.download_button('Download Model', f, file_name="best_model.pkl")