feat: use keras_image_embeddings_3_0 by default in categorize CLI

openfoodfacts · Mar 15, 2023 · 737716e · 737716e
1 parent a1bb507
commit 737716e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/robotoff/cli/main.py b/robotoff/cli/main.py
@@ -139,7 +139,8 @@ def categorize(
     barcode: str,
     deepest_only: bool = False,
     model_name: NeuralCategoryClassifierModel = typer.Option(
-        NeuralCategoryClassifierModel.keras_2_0, help="name of the model to use"
+        NeuralCategoryClassifierModel.keras_image_embeddings_3_0,
+        help="name of the model to use",
     ),
     threshold: Optional[float] = typer.Option(0.5, help="detection threshold to use"),
 ) -> None: