모델등록 수정

This commit is contained in:
2026-01-27 17:13:52 +09:00
parent da9da26796
commit 6ed679c8d6

View File

@@ -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());
}