File size: 679 Bytes
4ec72e6
 
 
d2cb493
 
ca47061
a7f8465
0fe6a90
d4ac851
4374108
0b78266
d2cb493
fa76574
 
4374108
0b78266
b63b73a
4ec72e6
 
2c55d4d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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()