From 9881ed59cec200ab58421908422852fb4b385531 Mon Sep 17 00:00:00 2001 From: cheetahadmin Date: Thu, 18 Sep 2025 04:09:42 +0000 Subject: [PATCH] Update 1/model.py --- 1/model.py | 47 +++++++++++++++++++++++++++-------------------- 1 file changed, 27 insertions(+), 20 deletions(-) diff --git a/1/model.py b/1/model.py index 6137dea..29c278b 100644 --- a/1/model.py +++ b/1/model.py @@ -1,28 +1,35 @@ -# model.py - 최소한의 구현 import triton_python_backend_utils as pb_utils import numpy as np +import json +import random +import string class TritonPythonModel: def initialize(self, args): - print("Model initialized") - + self.logger = pb_utils.Logger + self.model_name = args["model_name"] + self.logger.log_info(f"'{self.model_name}' 모델 초기화 완료") + + def execute(self, requests): - print(f"Received {len(requests)} requests") responses = [] - - for i, request in enumerate(requests): - print(f"Processing request {i}") - - # 고정된 응답 생성 - output_data = "Hello World" - output_tensor = pb_utils.Tensor("OUTPUT", np.array([output_data])) - - response = pb_utils.InferenceResponse(output_tensors=[output_tensor]) - responses.append(response) - print(f"Response {i} created") - - print("All responses ready") - return responses - + + for request in requests: + random_string = ''.join( + random.choice(string.ascii_letters + string.digits) for _ in range(16) + ) + self.logger.log_info(f"OUTPUT 출력:\n{random_string}") + + output_tensor = pb_utils.Tensor( + "OUTPUT", + np.array([random_string.encode("utf-8")], dtype=np.object_) + ) + + responses.append(pb_utils.InferenceResponse( + output_tensors=[output_tensor] + )) + + return responses + def finalize(self): - print("Model finalized") \ No newline at end of file + pass \ No newline at end of file