merve HF Staff commited on
Commit
9bc6745
·
1 Parent(s): 15d267c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ canvas = gr.inputs.Image(source="canvas", shape=(28,28))
13
  text = gr.outputs.Textbox()
14
  def infer(image):
15
  cls = np.argmax(model.predict(np.expand_dims(image, axis = 0)[:,:,:,1]))
16
- if cls == 9:
17
  output = "Death eater detected! 💀"
18
  else:
19
  cls = labels[cls]
 
13
  text = gr.outputs.Textbox()
14
  def infer(image):
15
  cls = np.argmax(model.predict(np.expand_dims(image, axis = 0)[:,:,:,1]))
16
+ if cls == 5:
17
  output = "Death eater detected! 💀"
18
  else:
19
  cls = labels[cls]