Don't show categories with less then 1% confidence

results.py

@@ -4,19 +4,20 @@ import pandas as pd
 from languages import en, ru
 
 
-def filter_results(results : pd.DataFrame, top_k=0.95):
+category_init = "label"
+certainty_init = "score"
+
+
+def filter_results(results: pd.DataFrame, top_k=0.95):
     certs = results[certainty_init].to_numpy()
     cum_certs = certs.cumsum()
     do_no_show = cum_certs < top_k
     first_not_to_show_id = np.argmin(do_no_show) + 1
     results = results.iloc[:first_not_to_show_id]
+    results = results[results[certainty_init] >= 0.01]
     return results
 
 
-category_init = "label"
-certainty_init = "score"
-
-
 def process_keys(results: pd.DataFrame, lang):
     category = {en: "Category", ru: "Категория"}
     certainty = {en: "Certainty", ru: "Уверенность"}
@@ -121,7 +122,7 @@ def process_results(results, lang):
     results = pd.DataFrame(results)
     results = process_categories(results, lang)
     results = results.groupby(by=category_init, as_index=False).sum()
-    results = results.sort_values(by=[certainty_init],ascending=False)
+    results = results.sort_values(by=[certainty_init], ascending=False)
     results = filter_results(results)
     results = process_certainities(results)
     results = process_keys(results, lang)