guru001's picture
should work now
a37501a
raw
history blame
371 Bytes
import numpy as np
import gradio as gr
from model import api
from PIL import Image
sign_api = api()
def sign(input_img):
input_img = Image.fromarray(input_img)
prediction = sign_api.predict(input_img)
print('prediction',prediction)
return prediction['class']
demo = gr.Interface(sign,inputs=gr.Image(shape=(200, 200)), outputs=gr.Label())
demo.launch()