init commit

app.py

@@ -0,0 +1,20 @@
+import tensorflow as tf
+import numpy as np
+from urllib.request import urlretrieve
+import gradio as gr
+
+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)}
+
+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.").launch();

requirements.txt

@@ -0,0 +1,2 @@
+tensorflow==2.3.0
+gradio>=2.2.6