Kalbe-x-Bangkit commited on
Commit
980ed1a
·
verified ·
1 Parent(s): 3c4bf46

Change model file name for detection.

Browse files

Change file name for detection from 'model.h5' into 'model-detection.h5'

Files changed (1) hide show
  1. app-streamlit.py +1 -1
app-streamlit.py CHANGED
@@ -32,7 +32,7 @@ W = 224
32
 
33
  @st.cache_resource
34
  def load_model():
35
- model = tf.keras.models.load_model("model.h5", compile=False)
36
  model.compile(
37
  loss={
38
  "bbox": "mse",
 
32
 
33
  @st.cache_resource
34
  def load_model():
35
+ model = tf.keras.models.load_model("model-detection.h5", compile=False)
36
  model.compile(
37
  loss={
38
  "bbox": "mse",