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']

css = ''

with gr.Blocks(css=css) as demo:
    gr.HTML("<h1><center>Signsapp: Classify the signs based on the hands sign images<center><h1>")
    gr.Interface(sign,inputs=gr.Image(shape=(200, 200)), outputs=gr.Label())

demo.launch()