Bug when loading keras model under keras2.3.0 and TensorFlow2.0

Summary

Bug when loading keras model under keras2.3.0 and TensorFlow2.0 if using st.button.

Steps to reproduce

import keras
import streamlit as st
def main():
    model = keras.applications.mobilenet.MobileNet(input_shape=None, alpha=1.0,
                                                   depth_multiplier=1, dropout=1e-3, include_top=True,
                                                   weights='imagenet', input_tensor=None, pooling=None, classes=1000)
    st.title("test")
    st.subheader(" test")
    st.subheader(" Extract from Your Text ")
    message = st.text_area(" Entre Your Text ", "  ")
    if st.button("Extract"):
        st.write(message)
if __name__ == '__main__':
    main()

What are the steps we should take to reproduce the bug:

1. load a keras model under keras2.3.0 and TensorFlow2.0.
2. streamlit run app.py
3. input some characters.
4. click the Extract button.
5. Error occurs. Any ideas? Thanks a lot.

AttributeError: ‘_thread._local’ object has no attribute ‘value’ Traceback: File “C:\Users\lxy\AppData\Roaming\Python\Python36\site-packages\streamlit\ScriptRunner.py”, line 311, in _run_script exec(code, module.dict)

Debug info

• Streamlit version: (get it with $ streamlit version) => 0.51.0
• Python version: (get it with $ python --version) => 3.6
• Using Conda? PipEnv? PyEnv? Pex? => Conda
• OS version: => win10
• Browser version: => chrome

[LCorleone]

@jrhone Thanks a lot. Hope someone can give me some suggestions!

[LCorleone]

@jrhone @monchier Thanks a lot. I think it is a special case I encounted. When i load a simple model, it is all ok. Maybe the complex model structure causes the error. It may be difficult to repro and i will not dig into this error. Close the issue.