Skip to content

Commit

Permalink
fix: add confidence score to image_flag prediction type
Browse files Browse the repository at this point in the history
The confidence score was only stored in data->likelihood field
  • Loading branch information
raphael0202 committed Mar 28, 2024
1 parent 651660d commit 94f151a
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions robotoff/prediction/ocr/image_flag.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 94f151a

Please sign in to comment.