Fix model API task type name classification->Classification (#1952)

Fix model_type str to have the value that ModelAPI expects.
openvinotoolkit · Mar 30, 2023 · 172dfd7 · 172dfd7
1 parent 6754fda
commit 172dfd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/otx/algorithms/classification/utils/cls_utils.py b/otx/algorithms/classification/utils/cls_utils.py
@@ -97,7 +97,7 @@ def get_cls_deploy_config(label_schema: LabelSchemaEntity, inference_config: Dic
 def get_cls_model_api_configuration(label_schema: LabelSchemaEntity, inference_config: Dict[str, Any]):
     """Get ModelAPI config."""
     mapi_config = {}
-    mapi_config[("model_info", "model_type")] = "classification"
+    mapi_config[("model_info", "model_type")] = "Classification"
     mapi_config[("model_info", "confidence_threshold")] = str(inference_config["confidence_threshold"])
     mapi_config[("model_info", "multilabel")] = str(inference_config["multilabel"])