Adapt confidence levels to new classification model

dnum-mi · Aug 28, 2023 · 18d91b3 · 18d91b3
1 parent 47ce1a3
commit 18d91b3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/src/main.py b/backend/src/main.py
@@ -276,9 +276,9 @@ async def imageupload(
         extras_logging["bg_label"] = label
         extras_logging["bg_confidence"] = confidence
         extras_logging["bg_model_time"] = round(time.time()-start, 2)
-        if confidence < 41:
+        if confidence < 46:
             extras_logging["bg_confidence_level"] = "low"
-        elif confidence < 65:
+        elif confidence < 76:
             extras_logging["bg_confidence_level"] = "medium"
         else:
             extras_logging["bg_confidence_level"] = "high"