diff --git a/robotoff/prediction/ocr/image_flag.py b/robotoff/prediction/ocr/image_flag.py index 41b50974c1..e69fa6e75e 100644 --- a/robotoff/prediction/ocr/image_flag.py +++ b/robotoff/prediction/ocr/image_flag.py @@ -127,6 +127,7 @@ def flag_image(content: Union[OCRResult, str]) -> list[Prediction]: "likelihood": label_annotation.score, }, predictor_version=PREDICTOR_VERSION, + confidence=label_annotation.score, ) ) break