Hugging Face's logo

Spaces:
SIB
/
violence_api
Sleeping

App Files Community
2
ZELEFACK commited on
Commit
2ba28bc
·
verified ·
1 Parent(s): 55260d6

, custom_objects={'CustomLayer': CustomLayer})

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import tensorflow as tf
3
- model_0 = tf.keras.models.load_model('bestmodel.h5')
4
  def classify_image(inp):
5
  inp = inp.reshape((-1, 224, 224, 3))
6
  prediction = model_0.predict(inp)
 
1
  import gradio as gr
2
  import tensorflow as tf
3
+ model_0 = tf.keras.models.load_model('bestmodel.h5', custom_objects={'CustomLayer': CustomLayer})
4
  def classify_image(inp):
5
  inp = inp.reshape((-1, 224, 224, 3))
6
  prediction = model_0.predict(inp)