Spaces:

Vallam1
/

mnist_cnn_binary

Runtime error

Vallam1 commited on Jul 2, 2022

Commit

2238089

•

1 Parent(s): 89c89e3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ from keras.models import load_model
 def cnn_model_binary(image):
   loaded_cnn_model = load_model("cnn_model_binary.h5")
-  prediction = loaded_cnn_model.predict(np.array([image]))
   return "NO" if prediction[0][0] != 1 else "YES"
 input_image = gr.inputs.Image(shape=(28, 28), image_mode='L')

 def cnn_model_binary(image):
   loaded_cnn_model = load_model("cnn_model_binary.h5")
+  prediction = loaded_cnn_model.predict(np.asarray([image]))
   return "NO" if prediction[0][0] != 1 else "YES"
 input_image = gr.inputs.Image(shape=(28, 28), image_mode='L')