File size: 611 Bytes
eda84c0 7b9be9f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import gradio as gr
from tensorflow import keras as k
import numpy as np
loaded_SNN = k.models.load_model('SequentialNN.h5')
def predict(img):
img_array = np.array(img)
img_array = np.array(img_array, dtype = 'int32').reshape(1, 28, 28)
pred = loaded_SNN.predict([img_array])
return np.argmax(pred)
iface = gr.Interface(predict, inputs = 'sketchpad',
outputs = 'text',
allow_flagging = 'never',
description = 'Draw a Digit Below... (Draw in the centre for best results)')
iface.launch(share = True, width = 500, height = 500) |