akhaliq HF Staff commited on
Commit
94b1f69
·
1 Parent(s): 6147796

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,9 +48,9 @@ def predict(path):
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
 
55
 
56
  title="DenseNet-121"
 
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]])
53
  return results
 
54
 
55
 
56
  title="DenseNet-121"