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 9f01f9a7..b4558d1d 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 @@ -132,11 +132,27 @@ public class InferenceResultService { Basic modelInfo = modelMngCoreService.findByModelUuid(uuid); + String cdModelPath = ""; + String cdModelConfigPath = ""; + String cdClsModelPath = ""; + + if (modelInfo.getCdModelPath() != null && modelInfo.getCdModelFileName() != null) { + cdModelPath = modelInfo.getCdModelPath() + "/" + modelInfo.getCdModelFileName(); + } + + if (modelInfo.getCdModelConfig() != null && modelInfo.getCdModelConfigFileName() != null) { + cdModelConfigPath = modelInfo.getCdModelConfig() + "/" + modelInfo.getCdModelConfigFileName(); + } + + if (modelInfo.getClsModelPath() != null && modelInfo.getClsModelFileName() != null) { + cdClsModelPath = modelInfo.getClsModelPath() + "/" + modelInfo.getClsModelFileName(); + } + InferenceSendDto sendDto = new InferenceSendDto(); sendDto.setModel_version(modelInfo.getModelVer()); - sendDto.setCd_model_path(modelInfo.getCdModelPath()); - sendDto.setCd_model_config(modelInfo.getCdModelConfig()); - sendDto.setCls_model_path(modelInfo.getClsModelPath()); + 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());