import gradio as gr from tensorflow import keras import numpy as np model = keras.models.load_model('Model') def greet(img): img = np.expand_dims(img, axis=0) return np.argmax(model.predict(img)[0]) demo = gr.Interface(fn=greet, inputs='sketchpad', outputs='text') demo.launch()