Sara Tolosa commited on
Commit
b0b0988
·
1 Parent(s): dd9445f

update environment

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ categories = ('Dog', 'Cat')
17
 
18
  def classify_image(img):
19
  # We are saying that this predictions returns: the prediction, its index and the prediction probability
20
- pred,idx,probs = learn.predict(img)
21
 
22
  # Here we return a dictionary with categories as keys and its probabilities as values
23
  return dict(zip(categories, map(float, probs)))
 
17
 
18
  def classify_image(img):
19
  # We are saying that this predictions returns: the prediction, its index and the prediction probability
20
+ pred,idx,probs = learner.predict(img)
21
 
22
  # Here we return a dictionary with categories as keys and its probabilities as values
23
  return dict(zip(categories, map(float, probs)))