fix(nutrisight): fix model dir

openfoodfacts · Dec 26, 2024 · 125250d · 125250d
1 parent 4f13fbb
commit 125250d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/robotoff/prediction/nutrition_extraction.py b/robotoff/prediction/nutrition_extraction.py
@@ -25,10 +25,10 @@
 logger = get_logger(__name__)
 
 MODEL_NAME = "nutrition_extractor"
-MODEL_VERSION = f"{MODEL_NAME}-1.0"
+MODEL_VERSION = f"{MODEL_NAME}-2.0"
 
 # The tokenizer assets are stored in the model directory
-MODEL_DIR = settings.TRITON_MODELS_DIR / f"{MODEL_NAME}/1/model.onnx"
+MODEL_DIR = settings.TRITON_MODELS_DIR / f"{MODEL_NAME}/2/model.onnx"
 
 
 @dataclasses.dataclass