IlyaHtuePav's picture
Update app.py
fe8d817
raw
history blame contribute delete
289 Bytes
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()