SalmanHabeeb commited on
Commit
16f509e
·
1 Parent(s): 0ede29c

Corrected variable name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ model = load_learner("model.pkl")
10
 
11
  def predict(img):
12
  pred, idx, probs = model.predict(img)
13
- dict1 = dict(zip(categories, map(float, preds)))
14
  dict1 = dict(sorted(dict1.items(), key=lambda item : item[1]))
15
  output = {key:dict1[key] for key in dict1.keys()[-3:]}
16
  sum = 0
 
10
 
11
  def predict(img):
12
  pred, idx, probs = model.predict(img)
13
+ dict1 = dict(zip(categories, map(float, probs)))
14
  dict1 = dict(sorted(dict1.items(), key=lambda item : item[1]))
15
  output = {key:dict1[key] for key in dict1.keys()[-3:]}
16
  sum = 0