모델등록 수정
This commit is contained in:
@@ -106,7 +106,7 @@ public class ModelMngService {
|
||||
modelUploadResDto.getClsModelFileName());
|
||||
|
||||
return new ApiResponseDto.ResponseObj(
|
||||
ApiResponseCode.NOT_FOUND, "pth,json,py파일이 모두 존재하지 않습니다");
|
||||
ApiResponseCode.NOT_FOUND, "pth,pt,py파일이 모두 존재하지 않습니다");
|
||||
}
|
||||
|
||||
addReq.setCdModelPath(modelUploadResDto.getCdModelPath());
|
||||
@@ -231,7 +231,7 @@ public class ModelMngService {
|
||||
int endPos = 20;
|
||||
|
||||
List<Basic> files =
|
||||
FIleChecker.getFilesFromAllDepth(dirPath, "*", "pth,py,json", 10, "name", startPos, endPos);
|
||||
FIleChecker.getFilesFromAllDepth(dirPath, "*", "pth,py,pt", 10, "name", startPos, endPos);
|
||||
|
||||
for (Basic dto : files) {
|
||||
// 예: 파일명 출력 및 추가 작업
|
||||
@@ -243,7 +243,7 @@ public class ModelMngService {
|
||||
} else if (dto.getExtension().equals("py")) {
|
||||
modelUploadResDto.setCdModelConfigPath(foldNm);
|
||||
modelUploadResDto.setCdModelConfigFileName(dto.getFileNm());
|
||||
} else if (dto.getExtension().equals("json")) {
|
||||
} else if (dto.getExtension().equals("pt")) {
|
||||
modelUploadResDto.setClsModelPath(foldNm);
|
||||
modelUploadResDto.setClsModelFileName(dto.getFileNm());
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user