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(model.predict(input_value / 255.))
gr.Interface(fn=predict_v,
            inputs = gr.SketchPad(),
            outputs="number").launch()