DurgeshRajput11 commited on
Commit
c859c2b
·
verified ·
1 Parent(s): fce3708

Update streamlit_app.py

Browse files
Files changed (1) hide show
  1. streamlit_app.py +1 -1
streamlit_app.py CHANGED
@@ -63,7 +63,7 @@ st.markdown("""
63
 
64
  @st.cache_resource
65
  def load_model():
66
- return tf.keras.models.load_model("trained_resnet50.pth")
67
 
68
  MODEL = load_model()
69
 
 
63
 
64
  @st.cache_resource
65
  def load_model():
66
+ return tf.keras.models.load_model("finetune_epoch_01_valacc_0.92.h5")
67
 
68
  MODEL = load_model()
69