fixed 95 percent choosing.

app.py

@@ -27,7 +27,7 @@ def is_local_machine():
 
 
 if is_local_machine():
-    model_path = os.path.expanduser("~/.cache/huggingface/checkpoints/distilbert-arxiv")
+    model_path = os.path.expanduser("~/.cache/huggingface/checkpoints/distilbert-arxiv2")
 else:
     model_path = "Hacker1337/distilbert-arxiv-checkpoint"
 
@@ -59,15 +59,14 @@ def infer(
     bar_plot_dict = {}
     total_prop = sum([prediction["score"] for prediction in predictions])
     gained_prob = 0
-    for prediction in predictions:
+    for prediction in sorted(predictions, key=lambda x: x["score"], reverse=True):
         bar_plot_dict[prediction["label"]] = prediction["score"]
-        if (gained_prob + prediction["score"]) / total_prop < target_probs_sum:
+        if (gained_prob) / total_prop < target_probs_sum:
             label_dict[category2human[prediction["label"]]] = (
                 prediction["score"] / total_prop
             )
             gained_prob += prediction["score"]
-
-    if gained_prob < total_prop:
+    if gained_prob < total_prop + 1e-5:
         label_dict["Other"] = (total_prop - gained_prob) / total_prop
     return df, label_dict