app.py

import gradio as gr]
import pickle
import numpy as np
model  = pickle.load(open("model.pkl","rb"))

def predict_v(input_value):
    try:
        if input_value == None: pass
    except:
        return np.argmax(input_value / 255.)
gr.Interface(fn=predict_v,
            inputs = gr.SketchPad(),
            outputs="number").launch()