fix: fix ImagePrediction.model_name

openfoodfacts · Jul 12, 2024 · 4a9e1a6 · 4a9e1a6
1 parent 18181bb
commit 4a9e1a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,