akhaliq HF Staff commited on
Commit
fd3b091
·
1 Parent(s): 54627b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def predict(path):
49
  a = np.argsort(preds)[::-1]
50
  results = {}
51
  for i in a[0:5]:
52
- results[labels[i]] = float(preds[0][0][i])
53
  return results
54
 
55
 
 
49
  a = np.argsort(preds)[::-1]
50
  results = {}
51
  for i in a[0:5]:
52
+ results[labels[a[i]]] = float(preds[a[i]])
53
  return results
54
 
55