akhaliq HF Staff commited on
Commit
fd8271e
·
1 Parent(s): 85e8652

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def predict(pil):
40
  {"data_0": img_batch_np.astype(np.float32)},
41
  )
42
 
43
- a = np.argsort(outputs[0].flatten())
44
  results = {}
45
  for i in a[0:5]:
46
  results[labels[i]]=float(outputs[0][0][i])
 
40
  {"data_0": img_batch_np.astype(np.float32)},
41
  )
42
 
43
+ a = np.argsort(-outputs[0].flatten())
44
  results = {}
45
  for i in a[0:5]:
46
  results[labels[i]]=float(outputs[0][0][i])