fix: fix category predictions

- the nutriment properties were not the right one in schema.py - we have to use V2 of Product Opener API to get nested ingredients, otherwise 3.0 models underperform
openfoodfacts · Feb 28, 2023 · 347c72c · 347c72c
1 parent f86c098
commit 347c72c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/robotoff/app/schema.py b/robotoff/app/schema.py
@@ -89,15 +89,15 @@
                         "nutriments": {
                             "type": "object",
                             "properties": {
-                                "fat": {"type": "number"},
-                                "saturated_fat": {"type": "number"},
-                                "carbohydrates": {"type": "number"},
-                                "sugars": {"type": "number"},
-                                "fiber": {"type": "number"},
-                                "proteins": {"type": "number"},
-                                "salt": {"type": "number"},
-                                "energy_kcal": {"type": "number"},
-                                "fruits_vegetables_nuts": {"type": "number"},
+                                "fat_100g": {"type": "number"},
+                                "saturated-fat_100g": {"type": "number"},
+                                "carbohydrates_100g": {"type": "number"},
+                                "sugars_100g": {"type": "number"},
+                                "fiber_100g": {"type": "number"},
+                                "proteins_100g": {"type": "number"},
+                                "salt_100g": {"type": "number"},
+                                "energy-kcal_100g": {"type": "number"},
+                                "fruits-vegetables-nuts_100g": {"type": "number"},
                             },
                             "additionalProperties": False,
                         },

diff --git a/robotoff/off.py b/robotoff/off.py
@@ -115,7 +115,9 @@ def get_product_image_select_url(server: Union[ServerType, str]) -> str:
 
 
 def get_api_product_url(server: Union[ServerType, str]) -> str:
-    return "{}/api/v0/product".format(get_base_url(server))
+    # V2 of API is required to have proper ingredient nesting
+    # for product categorization
+    return "{}/api/v2/product".format(get_base_url(server))
 
 
 def get_base_url(server: Union[ServerType, str]) -> str: