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")