diff --git a/app/service/generate_image/service_generate_product_image.py b/app/service/generate_image/service_generate_product_image.py index fa2340b..06f34d5 100644 --- a/app/service/generate_image/service_generate_product_image.py +++ b/app/service/generate_image/service_generate_product_image.py @@ -243,6 +243,8 @@ class GenerateProductImage: def callback(self, result, error): if error: self.gen_product_data['status'] = "FAILURE" + if 'mask_list' in str(error): + self.gen_product_data['status'] = "NO_FACE" self.gen_product_data['message'] = str(error) self.redis_client.set(self.tasks_id, json.dumps(self.gen_product_data)) else: diff --git a/app/service/generate_image/service_generate_relight_image.py b/app/service/generate_image/service_generate_relight_image.py index e0729ba..3efbe88 100644 --- a/app/service/generate_image/service_generate_relight_image.py +++ b/app/service/generate_image/service_generate_relight_image.py @@ -52,7 +52,11 @@ class GenerateRelightImage: def callback(self, result, error): if error: self.gen_product_data['status'] = "FAILURE" + if 'mask_list' in str(error): + self.gen_product_data['status'] = "NO_FACE" + self.gen_product_data['message'] = str(error) + self.redis_client.set(self.tasks_id, json.dumps(self.gen_product_data)) else: # pil图像转成numpy数组