Update app.py

app.py

@@ -2,14 +2,18 @@ import gradio as gr
 from huggingface_hub import from_pretrained_keras
 import numpy as np
 import logging
+from PIL import Image
 
 
 
  
 def fun(a):
+    Im=Image.fromarray(a).resize(48,48)
+    
     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=asarray(Im)
     a=a.reshape(1, 48, 48, 1)
     #reshape((-1,48,48,1))
     pred=reloaded_model.predict(a)
@@ -17,4 +21,4 @@ def fun(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()
+gr.Interface(fn=fun, inputs=gr.inputs.Image(image_mode='L',type='numpy',invert_colors=False), outputs=["text","text","text"]).launch()