Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 ==
|
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]
|