diff --git a/robotoff/insights/importer.py b/robotoff/insights/importer.py index 8a90cddd0b..4f4ef7b8e3 100644 --- a/robotoff/insights/importer.py +++ b/robotoff/insights/importer.py @@ -1222,7 +1222,8 @@ def get_nutrition_table_predictions( .where( ImageModel.barcode == product_id.barcode, ImageModel.server_type == product_id.server_type.name, - ImagePrediction.model_name == ObjectDetectionModel.get_type(), + ImagePrediction.model_name + == ObjectDetectionModel.nutrition_table_yolo.get_type(), ImagePrediction.max_confidence >= min_score, ) .tuples()