diff --git a/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java b/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java index d62ceeeb..d247920d 100644 --- a/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java +++ b/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java @@ -75,16 +75,6 @@ public class ModelMngService { addReq.setUuid(uuid); addReq.setClsModelVersion(addReq.getModelVer()); addReq.setPriority(0d); - Long modelUid = modelMngCoreService.insertModel(addReq); - - ModelMetricAddReq modelMetricAddReq = new ModelMetricAddReq(); - modelMetricAddReq.setModelUid(modelUid); - modelMetricAddReq.setModelVerUid(modelUid); - modelMetricAddReq.setF1Score(0); - modelMetricAddReq.setPrecision(0); - modelMetricAddReq.setRecall(0); - modelMetricAddReq.setLoss(0); - modelMetricAddReq.setIou(0); ModelUploadResDto modelUploadResDto = new ModelUploadResDto(); try { @@ -102,6 +92,19 @@ public class ModelMngService { throw new RuntimeException(e); } + Long modelUid = modelMngCoreService.insertModel(addReq); + + ModelMetricAddReq modelMetricAddReq = new ModelMetricAddReq(); + modelMetricAddReq.setModelUid(modelUid); + modelMetricAddReq.setModelVerUid(modelUid); + modelMetricAddReq.setF1Score(0); + modelMetricAddReq.setPrecision(0); + modelMetricAddReq.setRecall(0); + modelMetricAddReq.setLoss(0); + modelMetricAddReq.setIou(0); + + + ObjectMapper mapper = new ObjectMapper(); String filePath = modelUploadResDto.getClsModelPath() + modelUploadResDto.getClsModelFileName(); String dataJson = null;