jmparejaz commited on
Commit
f4f05ca
·
1 Parent(s): 0b78266

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,7 +18,7 @@ def fun(a):
18
  a=a.reshape((-1,48,48,1))
19
  pred=reloaded_model.predict(a)
20
  pred_eth=reloaded_model_eth.predict(a)
21
- return pred[0][0][0],pred[1][0][0],pred_eth
22
 
23
 
24
- gr.Interface(fn=fun, inputs=gr.inputs.Image(shape=(48,48)), outputs=["text","text","label"]).launch()
 
18
  a=a.reshape((-1,48,48,1))
19
  pred=reloaded_model.predict(a)
20
  pred_eth=reloaded_model_eth.predict(a)
21
+ return pred[0][0][0],pred[1][0][0],np.argmax(pred_eth)
22
 
23
 
24
+ gr.Interface(fn=fun, inputs=gr.inputs.Image(shape=(48,48)), outputs=["text","text","text"]).launch()