From ddb8145bd1d3d5e3d59e37e5df554fcb6d07a539 Mon Sep 17 00:00:00 2001
From: groupuser <groupuser@aifrica.co.kr>
Date: Tue, 29 Apr 2025 13:46:56 +0900
Subject: [PATCH] print file list at input model directory

---
 1/model.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/1/model.py b/1/model.py
index 52f3fb5..bbd0d27 100644
--- a/1/model.py
+++ b/1/model.py
@@ -22,6 +22,16 @@ class TritonPythonModel:
         #model_path = f"{model_repository}/{self.model_name}"
         model_path = "/cheetah/input/model/gemma-3-1b-it"
 
+        input_model_path = model_path
+
+        if os.path.exists(input_model_path):
+            file_list = os.listdir(input_model_path)
+            self.logger.log_info(f"'{input_model_path}' 디렉토리의 파일 목록:")
+        for file_name in file_list:
+            self.logger.log_info(file_name)
+        else:
+            self.logger.log_info(f"'{input_model_path}' 디렉토리가 존재하지 않습니다.")
+
         self.logger.log_info(f"model_repository: {model_repository}")
         self.logger.log_info(f"model_path: {model_path}")