Update app.py

app.py

@@ -1,8 +1,15 @@
 import gradio as gr
+import numpy as np
 from keras.models import load_model
 
+model = load_model("MNIST_model.keras")
+
 def MNIST(arr):
-    return arr.shape
+    arr = arr.reshape((1,28,28,1))
+    output = model(arr)
+    probs_list = [(val, idx) for idx, val in enumerate(probs_list[0])]
+    probs_list.sort(reverse=True)
+    return "\n".join([str(idx) + ": " + str(round((val * 100), 2)) + "% sure" for (val, idx) in probs_list])
 
 iface = gr.Interface(fn=MNIST, inputs="sketchpad", outputs="text")
 iface.launch()