fix: convert image to RGB before nutrition extraction (#1448)

openfoodfacts · Oct 28, 2024 · 96401ef · 96401ef
1 parent 022a788
commit 96401ef
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/robotoff/prediction/nutrition_extraction.py b/robotoff/prediction/nutrition_extraction.py
@@ -138,6 +138,10 @@ def preprocess(
     words = []
     char_offsets = []
     bboxes = []
+
+    if image.mode != "RGB":
+        image = image.convert("RGB")
+
     width, height = image.size
     for page in ocr_result.full_text_annotation.pages:
         for block in page.blocks: