import keras
import gradio as gr

model = keras.models.load_model("model.keras")


def recognize_digit(img):
    img = img.reshape(1, 784)
    img = img / 255
    prediction = model.predict(img).tolist()[0]
    return {str(i): prediction[i] for i in range(10)}


gr.Interface(
    recognize_digit,
    inputs=gr.Sketchpad(),
    outputs=gr.Label(),
    title="Predict from 0 to 9",
).launch()