auto-fl-fit / pages / 8_⬇️_Download.py
8_⬇️_Download.py
Raw
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")