handex commited on
Commit
3d523ad
·
1 Parent(s): 589232d

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -6,7 +6,7 @@ import numpy as np
6
 
7
  @st.cache_resource
8
  def load_my_model():
9
- model = tf.keras.models.load_model('src/cactus2.keras', compile=False)
10
  return model
11
 
12
  model=load_my_model()
 
6
 
7
  @st.cache_resource
8
  def load_my_model():
9
+ model = tf.keras.models.load_model("src/cactus2.keras",compile=False,safe_mode=False)
10
  return model
11
 
12
  model=load_my_model()