akhaliq HF staff commited on
Commit
7f941a2
1 Parent(s): f41d299

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def predict(path):
46
  ort_inputs = {session.get_inputs()[0].name: img}
47
  preds = session.run(None, ort_inputs)[0]
48
  preds = np.squeeze(preds)
49
- a = np.argsort(preds)[::1]
50
  results = {}
51
  for i in a[0:5]:
52
  results[labels[a[i]]] = float(preds[a[i]])
 
46
  ort_inputs = {session.get_inputs()[0].name: img}
47
  preds = session.run(None, ort_inputs)[0]
48
  preds = np.squeeze(preds)
49
+ a = np.argsort(preds)
50
  results = {}
51
  for i in a[0:5]:
52
  results[labels[a[i]]] = float(preds[a[i]])