diff --git a/robotoff/prediction/nutrition_extraction.py b/robotoff/prediction/nutrition_extraction.py index 1ff628e3b4..f965d7e5cd 100644 --- 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