diff --git a/app/service/brand_dna/service.py b/app/service/brand_dna/service.py index 012e682..5a525c2 100644 --- a/app/service/brand_dna/service.py +++ b/app/service/brand_dna/service.py @@ -9,7 +9,7 @@ import torch.nn.functional as F import tritonclient.http as httpclient from minio import Minio -from app.core.config import MINIO_URL, MINIO_ACCESS, MINIO_SECRET, MINIO_SECURE, DESIGN_MODEL_URL +from app.core.config import MINIO_URL, MINIO_ACCESS, MINIO_SECRET, MINIO_SECURE, DESIGN_MODEL_URL, CATEGORY_PATH from app.schemas.brand_dna import BrandDnaModel from app.service.attribute.config import local_debug_const from app.service.utils.generate_uuid import generate_uuid @@ -25,8 +25,8 @@ class BrandDna: self.sketch_bucket = "test" self.image_url = request_item.image_url self.is_brand_dna = request_item.is_brand_dna - # self.attr_type = pd.read_csv(CATEGORY_PATH) - self.attr_type = pd.read_csv(r"E:\workspace\trinity_client_aida\app\service\attribute\config\descriptor\category\category_dis.csv") + self.attr_type = pd.read_csv(CATEGORY_PATH) + # self.attr_type = pd.read_csv(r"E:\workspace\trinity_client_aida\app\service\attribute\config\descriptor\category\category_dis.csv") self.att_client = httpclient.InferenceServerClient(url=DESIGN_MODEL_URL) self.seg_client = httpclient.InferenceServerClient(url='10.1.1.243:30000') # self.const = const