jmparejaz commited on
Commit
a2191f3
·
1 Parent(s): fa76574

Update app.py

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