feat design相关模型迁移4090测试

fix
2024-11-08 14:05:09 +08:00
parent 6fe485f2ab
commit 13e3f8ac3d
4 changed files with 5 additions and 9 deletions
--- a/app/service/attribute/service_att_recognition.py
+++ b/app/service/attribute/service_att_recognition.py
@@ -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: