Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from keras.models import model_from_json
+from tensorflow.keras.preprocessing import image
 import numpy as np
 
 
@@ -25,5 +26,5 @@ def predict_age_gender(image):
     return {'Gender': ['Fmale' if gender_predict > 0.5 else 'Male'], 'Age': age_predict[0][0]}
 
 
-iface = gr.Interface(predict_age_gender, gr.Image(shape=(128, 128)), gr.Text())
+iface = gr.Interface(predict_age_gender, gr.Image(), gr.Text())
 iface.launch()