akhaliq HF Staff commited on
Commit
bdcf6dc
·
1 Parent(s): 977aca1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def predict(path):
52
  {"data_0": img_batch.astype(np.float32)},
53
  )
54
 
55
- a = np.argsort(outputs[0].flatten())
56
  results = {}
57
  for i in a[0:5]:
58
  results[labels[i]]=float(outputs[0][0][i])
 
52
  {"data_0": img_batch.astype(np.float32)},
53
  )
54
 
55
+ a = np.argsort(outputs[0].flatten())[::-1]
56
  results = {}
57
  for i in a[0:5]:
58
  results[labels[i]]=float(outputs[0][0][i])