dashasabirova commited on
Commit
dbf6476
1 Parent(s): f705779

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,6 +44,6 @@ indices = np.argsort(out_second)[0][::-1]
44
  sum_prob = 0
45
  for i in indices:
46
  print(labels_articles[i+1])
47
- sum_prob += out_second[i]
48
  if sum_prob > 0.95:
49
  break
 
44
  sum_prob = 0
45
  for i in indices:
46
  print(labels_articles[i+1])
47
+ sum_prob += out_second[0][i]
48
  if sum_prob > 0.95:
49
  break