Update app.py

app.py

@@ -1,17 +1,21 @@
-import gradio as gr
 import tensorflow as tf
 import numpy as np
+from urllib.request import urlretrieve
+import gradio as gr
 
-
-model = tf.keras.models.load_model("number_recognition_model_colab.keras")
+urlretrieve("https://gr-models.s3-us-west-2.amazonaws.com/mnist-model.h5", "mnist-model.h5")
+model = tf.keras.models.load_model("mnist-model.h5")
 
 def recognize_digit(image):
-    prediction = model.predict(np.reshape(image, (1, 28, 28))).tolist()[0]
+    image = image.reshape(1, -1)  # add a batch dimension
+    prediction = model.predict(image).tolist()[0]
     return {str(i): prediction[i] for i in range(10)}
 
-sketchpad = gr.Sketchpad(shape=(28, 28))
 gr.Interface(fn=recognize_digit, 
-             inputs=sketchpad,
-             outputs="label",
-             title="Handwritten Digits Classifier",
-             description="This app uses lenet5 for handwritten digits classification").launch()
+             inputs="sketchpad", 
+             outputs=gr.outputs.Label(num_top_classes=3),
+             live=True,
+             css=".footer {display:none !important}",
+             # title="MNIST Sketchpad",
+             description="Draw a number 0 through 9 on the sketchpad, and see predictions in real time.",
+             thumbnail="https://raw.githubusercontent.com/gradio-app/real-time-mnist/master/thumbnail2.png").launch();