diff --git a/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java b/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java index b4558d1d..5c44b513 100644 --- a/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java +++ b/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java @@ -148,14 +148,24 @@ public class InferenceResultService { cdClsModelPath = modelInfo.getClsModelPath() + "/" + modelInfo.getClsModelFileName(); } + String modelType = ""; + + if (modelInfo.getModelType().equals("M1")) { + modelType = "G1"; + } else if (modelInfo.getModelType().equals("M2")) { + modelType = "G2"; + } else { + modelType = "G3"; + } + InferenceSendDto sendDto = new InferenceSendDto(); sendDto.setModel_version(modelInfo.getModelVer()); sendDto.setCd_model_path(cdModelPath); sendDto.setCd_model_config(cdModelConfigPath); sendDto.setCls_model_path(cdClsModelPath); - // sendDto.setCls_model_version(modelInfo.getCls); - sendDto.setCd_model_type(modelInfo.getModelType()); - // sendDto.setPriority(modelInfo.getPriority()); + sendDto.setCls_model_version(modelInfo.getModelVer()); + sendDto.setCd_model_type(modelType); + sendDto.setPriority(modelInfo.getPriority().intValue()); return sendDto; }