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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -6,17 +6,19 @@ import logging
6
 
7
 
8
 
9
- #reloaded_model_eth = from_pretrained_keras('jmparejaz/Facial_eth_recognition')
10
 
11
  def rgb2gray(rgb):
12
  return np.dot(rgb[...,:3], [0.2989, 0.5870, 0.1140])
13
 
14
  def fun(a):
15
  reloaded_model = from_pretrained_keras('jmparejaz/Facial_Age-gender-eth_Recognition')
 
16
  #img=load_img(a, grayscale=True)
17
  a=a.reshape((-1,48,48,1))
18
  pred=reloaded_model.predict(a)
19
- return pred[0][0][0],pred[1][0][0],
 
20
 
21
 
22
- gr.Interface(fn=fun, inputs=gr.inputs.Image(shape=(48,48)), outputs=["text","text"]).launch()
 
6
 
7
 
8
 
9
+ #
10
 
11
  def rgb2gray(rgb):
12
  return np.dot(rgb[...,:3], [0.2989, 0.5870, 0.1140])
13
 
14
  def fun(a):
15
  reloaded_model = from_pretrained_keras('jmparejaz/Facial_Age-gender-eth_Recognition')
16
+ reloaded_model_eth = from_pretrained_keras('jmparejaz/Facial_eth_recognition')
17
  #img=load_img(a, grayscale=True)
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()