Merge pull request #221 from computas/SP24-798-better-categories

making labels appear less times
computas · Oct 10, 2024 · 59b894f · 59b894f
2 parents d215236 + aef1cf5
commit 59b894f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/customvision/classifier.py b/src/customvision/classifier.py
@@ -84,7 +84,10 @@ def __init__(self) -> None:
             puplished_iterations.sort(key=lambda i: i.created)
             self.iteration_name = puplished_iterations[-1].publish_name
         except Exception as e:
-            logging.info("An error occurred while trying to get latest published iteration from Custom Vision", e)
+            logging.info(
+                "An error occurred while trying to get latest published iteration from Custom Vision",
+                e,
+            )
             self.iteration_name = "Iteration4"
 
     def predict_image_url(self, img_url: str) -> Dict[str, float]:

diff --git a/src/models.py b/src/models.py
@@ -540,7 +540,7 @@ def get_n_labels(n, difficulty_id):
                 difficulty_id - Games.difficulty_id < 2,
             )
             .order_by(Games.date.desc())
-            .limit(3)
+            .limit(5)
             .all()
         )
         labels_to_filter = [