jmparejaz's picture
Update app.py
2c55d4d
raw
history blame
679 Bytes
import gradio as gr
from huggingface_hub import from_pretrained_keras
import numpy as np
import logging
def fun(a):
reloaded_model = from_pretrained_keras('jmparejaz/Facial_Age-gender-eth_Recognition')
reloaded_model_eth = from_pretrained_keras('jmparejaz/Facial_eth_recognition')
#img=load_img(a, grayscale=True)
a=a.reshape(1, 48, 48, 1)
#reshape((-1,48,48,1))
pred=reloaded_model.predict(a)
pred_eth=reloaded_model_eth.predict(a)
return pred[0][0][0],pred[1][0][0],np.argmax(pred_eth)
gr.Interface(fn=fun, inputs=gr.inputs.Image(shape=(48,48),image_mode='L',type='numpy',invert_colors=False), outputs=["text","text","text"]).launch()