sensei-ml's picture
Update app.py to implement the classification function
bcb272d
raw
history blame
213 Bytes
import gradio as gr
from model import model
def classify_image(image):
result = model.predict(image)
return result
demo = gr.Interface(fn=classify_image, inputs=gr.Image(), outputs="text")
demo.launch()