Update confidence levels values

dnum-mi · Jan 22, 2024 · fb101c7 · fb101c7
1 parent 3d41b0e
commit fb101c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/src/main.py b/backend/src/main.py
@@ -242,9 +242,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 < 0.46:
+        if confidence < 0.76:
             extras_logging["bg_confidence_level"] = "low"
-        elif confidence < 0.76:
+        elif confidence < 0.99:
             extras_logging["bg_confidence_level"] = "medium"
         else:
             extras_logging["bg_confidence_level"] = "high"