File size: 298 Bytes
eaacb7e 5624821 eaacb7e | 1 2 3 4 5 6 7 8 9 10 | import gradio as gr
import numpy as np
from tensorflow import keras
model = keras.models.load_model("my_new_model")
def greet(img):
img = np.expand_dims(img, axis = 0)
return np.argmax(model.predict(img)[0])
iface = gr.Interface(fn=greet, inputs="sketchpad", outputs="text")
iface.launch() |