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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
6
 
7
  def input_img(img):
8
  learn=load_learner('model.pkl')
9
- race,_,probs = learn.predict(PILImage.create(img))
10
  #print(f"This is a: {race}.")
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
 
6
 
7
  def input_img(img):
8
  learn=load_learner('model.pkl')
9
+ race,_,probs = learn.predict(img)
10
  #print(f"This is a: {race}.")
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