추론실행 수정

src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java

@@ -56,6 +56,12 @@ public class ModelMngService {
   @Value("${file.model-tmp-dir}")
   private String modelTmpDir;
 
+  @Value("${file.pt-path}")
+  private String ptPath;
+
+  @Value("${file.pt-FileName}")
+  private String ptFileName;
+
   public Page<ModelMngDto.ModelList> findModelMgmtList(
       ModelMngDto.searchReq searchReq,
       LocalDate startDate,
@@ -236,7 +242,7 @@ public class ModelMngService {
 
     List<Basic> files =
         FIleChecker.getFilesFromAllDepth(
-            dirPath, "*", "pth,py,pt,json", 10, "name", startPos, endPos);
+            dirPath, "*", "pth,py,json", 10, "name", startPos, endPos);
 
     boolean hasPt = false; // pt 파일 존재 여부
 
@@ -253,11 +259,6 @@ public class ModelMngService {
           modelUploadResDto.setCdModelConfigPath(foldNm);
           modelUploadResDto.setCdModelConfigFileName(dto.getFileNm());
         }
-        case "pt" -> {
-          modelUploadResDto.setClsModelPath(foldNm);
-          modelUploadResDto.setClsModelFileName(dto.getFileNm());
-          hasPt = true;
-        }
         case "json" -> {
           modelUploadResDto.setJsonPath(foldNm);
           modelUploadResDto.setJsonFileName(dto.getFileNm());
@@ -265,18 +266,10 @@ public class ModelMngService {
       }
     }
 
-    if (!hasPt) {
+    // pt는 고정경로 등록
+    modelUploadResDto.setClsModelPath(ptPath);
+    modelUploadResDto.setClsModelFileName(ptFileName);
 
-      String defaultPath = "/kamco-nfs/ckpt/classification/";
-      String defaultFileName = "v5-best.pt";
-
-      Path ptPath = Paths.get(defaultPath, defaultFileName);
-
-      if (Files.exists(ptPath)) {
-        modelUploadResDto.setClsModelPath(defaultPath);
-        modelUploadResDto.setClsModelFileName(defaultFileName);
-      }
-    }
 
     // int fileListPos = 0;
     // int fileTotCnt = files.size();