Update app.py

app.py

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