junk727/web_gui_wrapper

1import gradio as gr2import numpy as np3import keras4import os5 6model = keras.models.load_model("mnist_model.h5")7 8def rgb2gray(rgb):9 10    return [255] - np.dot(rgb[..., :3], [0.2989, 0.5870, 0.1140])11 12def number_classifier(target):13 14    target = rgb2gray(target).flatten()15 16    inputs = np.array([target])17 18    results = model.predict(inputs)19 20    result_as_dict = {}21 22    for i in range(10):23 24        result_as_dict[str(i)] = float(results[0][i])25 26    return result_as_dict27 28examples_list = []29 30for item in os.listdir("examples/"):31    examples_list.append("examples/" + item)32 33app = gr.Interface(fn=number_classifier,34                   inputs=[gr.Image(shape=(28, 28))],35                   outputs=[gr.Label(num_top_classes=3)],36                   examples=examples_list37                   )38 39app.launch()