zferd commited on
Commit
2d9bc18
·
verified ·
1 Parent(s): ec26494

Upload streamlit_app.py

Browse files
Files changed (1) hide show
  1. streamlit_app.py +1 -1
streamlit_app.py CHANGED
@@ -35,7 +35,7 @@ if 'probs' not in st.session_state:
35
 
36
  # ---- Hugging Face Repo Details --- #
37
  REPO_ID = "zferd/welding-defect"
38
- MODEL_FILENAME = "model/final_single_phase.keras"
39
  CONFIG_FILENAME = "model/training_config.json"
40
  IMG_SIZE = (224, 224)
41
 
 
35
 
36
  # ---- Hugging Face Repo Details --- #
37
  REPO_ID = "zferd/welding-defect"
38
+ MODEL_FILENAME = "model/final_single_phase.h5"
39
  CONFIG_FILENAME = "model/training_config.json"
40
  IMG_SIZE = (224, 224)
41