import tensorflow as tf
import numpy as np
from urllib.request import urlretrieve
import gradio as gr
import numpy as np

urlretrieve("https://huggingface.co/guiwitz/mnist2023/resolve/main/mnist_model.keras", "mnist_model.keras")
model = tf.keras.models.load_model("mnist_model.keras")

def recognize_digit(image):
    image = image[np.newaxis,:,:, np.newaxis]
    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.Label(num_top_classes=3),
             live=True,
             description="Live MNIST.").launch();