akhaliq HF Staff commited on
Commit
58659b1
·
1 Parent(s): 152fb4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -46,9 +46,10 @@ 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
- results[labels[a[0]]] = float(preds[a[0]]*0.1)
 
52
  return results
53
 
54
 
 
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[0]]] = float(preds[a[0]])
53
  return results
54
 
55