HoaAn2003 commited on
Commit
050df53
·
1 Parent(s): 82ed154

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ with open('categories.txt', 'r') as file:
7
  categories = file.read().split()
8
 
9
  def classify_image(image):
10
- pred,idx,probs = learner.predict()
11
  return dict(zip(categories,map(float,probs)))
12
 
13
  ins = gr.inputs.Image()
 
7
  categories = file.read().split()
8
 
9
  def classify_image(image):
10
+ pred,idx,probs = learner.predict(image)
11
  return dict(zip(categories,map(float,probs)))
12
 
13
  ins = gr.inputs.Image()