Kalbe-x-Bangkit
commited on
Change model file name for detection.
Browse filesChange file name for detection from 'model.h5' into 'model-detection.h5'
- 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",
|