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