Update app.py

app.py

@@ -51,6 +51,7 @@ def top_xx(preds, xx=95):
     while total < xx / 100:
         next_id = tops[0, index].item()
         if next_id == 155:
+            index += 1
             continue
         total += preds[0, next_id]
         index += 1