From c96752e59929b4c683219cddd1ab96ec3883a45b Mon Sep 17 00:00:00 2001
From: felixm <mail@felixm.de>
Date: Sat, 1 Jul 2023 13:17:39 +0200
Subject: [PATCH] Fix sort order

---
 src/predict.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/predict.py b/src/predict.py
index 8482def..40e945b 100644
--- a/src/predict.py
+++ b/src/predict.py
@@ -15,12 +15,12 @@ def get_sort_categories():
             return
         _, _, probs = learn.predict(row)
         cat_to_prob = dict(zip(learn.dls.vocab[1],probs.tolist()))
-        categories.sort(key=lambda c: cat_to_prob[c] if c in cat_to_prob else 0.0)
+        categories.sort(key=lambda c: cat_to_prob[c] if c in cat_to_prob else 0.0, reverse=True)
 
     learn = None
     try:
         from fastai.text.all import load_learner
-        learn = load_learner("ldg.pkl")
+        learn = load_learner("export.pkl")
     except ModuleNotFoundError:
         pass
     except FileNotFoundError: