feat 2倍超分

2024-04-11 14:55:06 +08:00
parent fbb246d0de
commit 284ab0f2a0
1 changed files with 1 additions and 0 deletions
--- a/app/service/super_resolution/service.py
+++ b/app/service/super_resolution/service.py
@@ -27,6 +27,7 @@ logger = logging.getLogger()

 class SuperResolution:
    def __init__(self, data):
+        logger.info(f"sr triton service url is : {SR_TRITON_URL}")
        self.triton_client = grpcclient.InferenceServerClient(url=SR_TRITON_URL)
        self.redis_client = redis.StrictRedis(host=REDIS_HOST, port=REDIS_PORT, db=REDIS_DB, decode_responses=True)
        self.tasks_id = data.sr_tasks_id