jmparejaz's picture
Update app.py
faa7830
raw
history blame
531 Bytes
import gradio as gr
from huggingface_hub import from_pretrained_keras
import numpy as np
from PIL import Image
#reloaded_model_eth = from_pretrained_keras('jmparejaz/Facial_eth_recognition')
def rgb2gray(rgb):
return np.dot(rgb[...,:3], [0.2989, 0.5870, 0.1140])
def fun(a):
reloaded_model = from_pretrained_keras('jmparejaz/Facial_Age-gender-eth_Recognition')
inp = a.reshape((-1, 48, 48, 3))
return Image.fromarray(inp)
gr.Interface(fn=fun, inputs=gr.Image(shape=(48, 48)), outputs="Image").launch()