diff --git a/robotoff/insights/extraction.py b/robotoff/insights/extraction.py index dbdb978e42..5f2ac4016d 100644 --- a/robotoff/insights/extraction.py +++ b/robotoff/insights/extraction.py @@ -93,7 +93,7 @@ def run_object_detection_model( return ImagePrediction.create( image=image_model, type="object_detection", - model_name=model_name.value, + model_name=model_name.get_type(), model_version=OBJECT_DETECTION_MODEL_VERSION[model_name], data={"objects": data}, timestamp=timestamp,