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