testing by adding custom_objects={'DepthwiseConv2D': tf.keras.layers.DepthwiseConv2D}
Browse files
app.py
CHANGED
|
@@ -332,7 +332,7 @@ def redirect_button(url):
|
|
| 332 |
st.markdown(f'<meta http-equiv="refresh" content="0;url={url}" />', unsafe_allow_html=True)
|
| 333 |
|
| 334 |
def load_model():
|
| 335 |
-
model = tf.keras.models.load_model('./model.h5')
|
| 336 |
return model
|
| 337 |
|
| 338 |
###########################################################################################
|
|
|
|
| 332 |
st.markdown(f'<meta http-equiv="refresh" content="0;url={url}" />', unsafe_allow_html=True)
|
| 333 |
|
| 334 |
def load_model():
|
| 335 |
+
model = tf.keras.models.load_model('./model.h5',custom_objects={'DepthwiseConv2D': tf.keras.layers.DepthwiseConv2D})
|
| 336 |
return model
|
| 337 |
|
| 338 |
###########################################################################################
|