Update app.py

app.py

@@ -7,9 +7,7 @@ model = load_model("MNIST_model.keras")
 def MNIST(arr):
     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])
+    return {idx: float(val) for idx, val in enumerate(output[0])}
 
-iface = gr.Interface(fn=MNIST, inputs="sketchpad", outputs="text")
+iface = gr.Interface(fn=MNIST, inputs="sketchpad", outputs="label")
 iface.launch()