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 |
###########################################################################################
|