From 2226bf03c02795e62828f8c012094ad179f7840e Mon Sep 17 00:00:00 2001 From: cheetahadmin Date: Thu, 18 Sep 2025 05:19:18 +0000 Subject: [PATCH] Update 1/model.py --- 1/model.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/1/model.py b/1/model.py index 29c278b..cf91943 100644 --- a/1/model.py +++ b/1/model.py @@ -7,18 +7,21 @@ import string class TritonPythonModel: def initialize(self, args): self.logger = pb_utils.Logger - self.model_name = args["model_name"] - self.logger.log_info(f"'{self.model_name}' 모델 초기화 완료") - + self.logger.log_info(f"'{args["model_name"]}' 모델 초기화 완료") def execute(self, requests): responses = [] for request in requests: + input_tensor = pb_utils.get_input_tensor_by_name(request, "INPUT") + + input_data = input_tensor.as_numpy()[0].decode("utf-8") + self.logger.log_info(f"INPUT: {input_data}") + random_string = ''.join( - random.choice(string.ascii_letters + string.digits) for _ in range(16) + random.choice(string.ascii_letters + string.digits) for _ in range(10) ) - self.logger.log_info(f"OUTPUT 출력:\n{random_string}") + self.logger.log_info(f"OUTPUT: {random_string}") output_tensor = pb_utils.Tensor( "OUTPUT", @@ -29,7 +32,7 @@ class TritonPythonModel: output_tensors=[output_tensor] )) - return responses + return responses def finalize(self): pass \ No newline at end of file