farid678 commited on
Commit
cffe539
1 Parent(s): 4a4a28f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,5 +25,5 @@ def predict_age_gender(image):
25
  return {'Gender': ['Fmale' if gender_predict > 0.5 else 'Male'], 'Age': age_predict[0][0]}
26
 
27
 
28
- iface = gr.Interface(predict_age_gender, gr.Image(), gr.Text())
29
  iface.launch()
 
25
  return {'Gender': ['Fmale' if gender_predict > 0.5 else 'Male'], 'Age': age_predict[0][0]}
26
 
27
 
28
+ iface = gr.Interface(predict_age_gender, gr.Image(shape=(128, 128)), gr.Text())
29
  iface.launch()