Spaces:
Runtime error
Runtime error
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) | |
return a.shape,a[0][0] | |
gr.Interface(fn=fun, inputs=gr.inputs.Image(shape=(48,48)), outputs=["text","text","text"]).launch() |