aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat 2倍超分

Browse Source
This commit is contained in:
zhouchengrong
2024-04-11 14:55:06 +08:00
parent fbb246d0de
commit 284ab0f2a0
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app/service/super_resolution/service.py
View File

@@ -27,6 +27,7 @@ logger = logging.getLogger()
class SuperResolution: class SuperResolution:
def __init__(self, data): 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.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.redis_client = redis.StrictRedis(host=REDIS_HOST, port=REDIS_PORT, db=REDIS_DB, decode_responses=True)
self.tasks_id = data.sr_tasks_id self.tasks_id = data.sr_tasks_id