The secret is to use @st.cache(ignore_hash=True) to store Tensorflow state. If more questions / issues come up, please post them here or in our forums discuss.streamlit.io.