feat/infer_dev_260107 #1
@@ -106,7 +106,7 @@ public class ModelMngService {
|
|||||||
modelUploadResDto.getClsModelFileName());
|
modelUploadResDto.getClsModelFileName());
|
||||||
|
|
||||||
return new ApiResponseDto.ResponseObj(
|
return new ApiResponseDto.ResponseObj(
|
||||||
ApiResponseCode.NOT_FOUND, "pth,json,py파일이 모두 존재하지 않습니다");
|
ApiResponseCode.NOT_FOUND, "pth,pt,py파일이 모두 존재하지 않습니다");
|
||||||
}
|
}
|
||||||
|
|
||||||
addReq.setCdModelPath(modelUploadResDto.getCdModelPath());
|
addReq.setCdModelPath(modelUploadResDto.getCdModelPath());
|
||||||
@@ -231,7 +231,7 @@ public class ModelMngService {
|
|||||||
int endPos = 20;
|
int endPos = 20;
|
||||||
|
|
||||||
List<Basic> files =
|
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) {
|
for (Basic dto : files) {
|
||||||
// 예: 파일명 출력 및 추가 작업
|
// 예: 파일명 출력 및 추가 작업
|
||||||
@@ -243,7 +243,7 @@ public class ModelMngService {
|
|||||||
} else if (dto.getExtension().equals("py")) {
|
} else if (dto.getExtension().equals("py")) {
|
||||||
modelUploadResDto.setCdModelConfigPath(foldNm);
|
modelUploadResDto.setCdModelConfigPath(foldNm);
|
||||||
modelUploadResDto.setCdModelConfigFileName(dto.getFileNm());
|
modelUploadResDto.setCdModelConfigFileName(dto.getFileNm());
|
||||||
} else if (dto.getExtension().equals("json")) {
|
} else if (dto.getExtension().equals("pt")) {
|
||||||
modelUploadResDto.setClsModelPath(foldNm);
|
modelUploadResDto.setClsModelPath(foldNm);
|
||||||
modelUploadResDto.setClsModelFileName(dto.getFileNm());
|
modelUploadResDto.setClsModelFileName(dto.getFileNm());
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user