Update app.py

app.py

@@ -19,7 +19,10 @@ def fun(a):
     #reshape((-1,48,48,1))
     pred=reloaded_model.predict(a)
     pred_eth=reloaded_model_eth.predict(a)
-    return pred[0][0][0],pred[1][0][0],np.argmax(pred_eth)
+    dict_gender={0:'Male',1:'Female'}
+    dict_eth={0:"White", 1:"Black", 2:"Asian", 3:"Indian", 4:"Hispanic"}
+    
+    return dict_gender[pred[0][0][0]],np.round(pred[1][0][0]),dict_eth[np.argmax(pred_eth)]
 
 
 gr.Interface(fn=fun, inputs=gr.inputs.Image(image_mode='L',type='numpy',invert_colors=False), outputs=["text","text","text"]).launch()