Update app.py

app.py

@@ -15,8 +15,10 @@ def predict_age_gender(image):
     
     model = model_from_json(loaded_file_json)
     model.load_weights('xception_modelv2.h5')
-    img_pixels = np.expand_dims(image, axis=0)
-    img_pixels = image.astype('float')
+
+    images = np.resize(image, new_shape=(1, 128, 128, 3))
+    img_pixels = np.expand_dims(images, axis=0)
+    img_pixels = images.astype('float')
     img_pixels = img_pixels.reshape((1, 128, 128, 3))
     img_pixels /= 255
 
@@ -26,5 +28,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()