File size: 291 Bytes
5600cc5 f904e8f 5600cc5 3862d92 5600cc5 3862d92 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | import numpy as np
import gradio as gr
from tensorflow import keras
model = keras.models.load_model('Abra')
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() |