Update app.py

app.py

@@ -13,7 +13,7 @@ def recognize_digit(image):
   if image is not None:
     image = image.reshape((1,28,28,1)).astype('float32')/255
     prediction = model.predict(image)
-    return {str(i) : float(prediction[0][i]) for i in range(2)}
+    return {str(i) : float(prediction[0][i]) for i in range(10)}
   else:
     return ''