aidlab/sora_python
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新一些配置与配置相关的代码

Browse Source
This commit is contained in:
zhouchengrong
2024-03-18 17:57:01 +08:00
parent 8383234ba1
commit f84b8aaba1
6 changed files with 71 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/service/attribute_recognition/service.py
View File

@@ -13,7 +13,7 @@ from torchvision import transforms
import pandas as pd
from minio import Minio
from app.core.config import MINIO_IP, MINIO_PORT, MINIO_SECURE, MINIO_ACCESS, MINIO_SECRET
from app.core.config import MINIO_IP, MINIO_PORT, MINIO_SECURE, MINIO_ACCESS, MINIO_SECRET, ATT_TRITON_IP, ATT_TRITON_PORT
def Merge(dict1, dict2):
@@ -54,7 +54,7 @@ class Rescale(object):
class AttributeRecognition:
def __init__(self):
self.httpclient = httpclient.InferenceServerClient(url="10.1.1.150:6000")
self.httpclient = httpclient.InferenceServerClient(url=f"{ATT_TRITON_IP}:{ATT_TRITON_PORT}")
self.minio_client = Minio(
f"{MINIO_IP}:{MINIO_PORT}",
access_key=MINIO_ACCESS,