import gradio as gr
from model import model
import numpy as np


def predict_age_gender(image):
    img_pixels = np.expand_dims(image, axis=0)
    img_pixels = image.astype('float')
    img_pixels = img_pixels.reshape((1, 128, 128, 3))
    img_pixels /= 255

    predict = model.predict(img_pixels)
    gender_predict = predict[0]
    age_predict = predict[1]
    return {'Gender': ['Fmale' if gender_predict > 0.5 else 'Male'], 'Age': age_predict[0][0]}


iface = gr.Interface(predict_age_gender, gr.Image(), gr.Text())
iface.launch(share=True)