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

feat design相关模型迁移4090测试

Browse Source 
fix
This commit is contained in:
zhouchengrong
2024-11-08 14:05:09 +08:00
parent 0a17128140
commit d92c59383b
4 changed files with 5 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/service/attribute/service_att_recognition.py
View File

@@ -28,7 +28,7 @@ class AttributeRecognition:
}
)
self.const = const
self.triton_client = httpclient.InferenceServerClient(url=f"{ATT_TRITON_URL}")
self.triton_client = httpclient.InferenceServerClient(url=f"{DESIGN_MODEL_URL}")
def get_result(self):
for sketch in self.request_data:

2
app/service/attribute/service_category_recognition.py
View File

@@ -26,7 +26,7 @@ class CategoryRecognition:
self.attr_type = pd.read_csv(CATEGORY_PATH)
# self.minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
self.request_data = []
self.triton_client = httpclient.InferenceServerClient(url=ATT_TRITON_URL)
self.triton_client = httpclient.InferenceServerClient(url=DESIGN_MODEL_URL)
for sketch in request_data:
self.request_data.append(
{