Kalbe-x-Bangkit
commited on
Commit
•
528572f
1
Parent(s):
d1b31a5
Update app.py
Browse files
app.py
CHANGED
@@ -371,7 +371,7 @@ def redirect_button(url):
|
|
371 |
st.markdown(f'<meta http-equiv="refresh" content="0;url={url}" />', unsafe_allow_html=True)
|
372 |
|
373 |
def load_model():
|
374 |
-
model = tf.keras.models.load_model('./
|
375 |
return model
|
376 |
|
377 |
###########################################################################################
|
|
|
371 |
st.markdown(f'<meta http-equiv="refresh" content="0;url={url}" />', unsafe_allow_html=True)
|
372 |
|
373 |
def load_model():
|
374 |
+
model = tf.keras.models.load_model('./model_renamed.h5',custom_objects={'DepthwiseConv2D': tf.keras.layers.DepthwiseConv2D})
|
375 |
return model
|
376 |
|
377 |
###########################################################################################
|