Update app.py

app.py

@@ -3,15 +3,16 @@ Author       : Alafun
 GitHub Page  : https://github.com/Alafun/
 Date         : 2022-06-21 16:43:21
 LastEditors  : Alafun
-LastEditTime : 2022-06-21 16:45:11
+LastEditTime : 2022-06-21 18:51:37
 Description  : 
 FilePath     : \\app.py
 Copyright (c) 2022 by Alafun, All Rights Reserved. 
 '''
-import numpy as np
+
 
 import gradio as gr
 
+
 def sepia(input_img):
     sepia_filter = np.array(
         [[0.393, 0.769, 0.189], [0.349, 0.686, 0.168], [0.272, 0.534, 0.131]]
@@ -20,6 +21,27 @@ def sepia(input_img):
     sepia_img /= sepia_img.max()
     return sepia_img
 
-demo = gr.Interface(sepia, gr.Image(shape=(200, 200)), "image")
+import tensorflow as tf
+import numpy as np
+from urllib.request import urlretrieve
+
+
+
+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):
+    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)}
+
+demo = gr.Interface(fn=recognize_digit,
+                    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")
 
-demo.launch()
+demo.launch()