rafaym commited on
Commit
3ea2109
·
1 Parent(s): 78230ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -11,10 +11,10 @@ def input_img(img):
11
  processed_output=(f"This is a: {race}./nProbability it's a black person: {probs[0]:.4f}.\nProbability it's a white person: {probs[1]:.4f}")
12
  return processed_output
13
 
14
- categories=('Black people','White people')
15
- def func_classi(img):
16
- pred,idx,probs=learn.predict(img)
17
- return dict(zip(categories,map(float,probs)))
18
 
19
  image=gr.inputs.Image(shape=(192,192))
20
  label=gr.outputs.Label()
 
11
  processed_output=(f"This is a: {race}./nProbability it's a black person: {probs[0]:.4f}.\nProbability it's a white person: {probs[1]:.4f}")
12
  return processed_output
13
 
14
+ # categories=('Black people','White people')
15
+ # def func_classi(img):
16
+ # pred,idx,probs=learn.predict(img)
17
+ # return dict(zip(categories,map(float,probs)))
18
 
19
  image=gr.inputs.Image(shape=(192,192))
20
  label=gr.outputs.Label()