diff --git a/app/service/generate_image/service_generate_product_image.py b/app/service/generate_image/service_generate_product_image.py index 06f34d5..235f366 100644 --- a/app/service/generate_image/service_generate_product_image.py +++ b/app/service/generate_image/service_generate_product_image.py @@ -302,7 +302,7 @@ class GenerateProductImage: time_out = 600 while time_out > 0: gen_product_data, _ = self.read_tasks_status() - if gen_product_data['status'] in ["REVOKED", "FAILURE"]: + if gen_product_data['status'] in ["REVOKED", "FAILURE", "NO_FACE"]: ctx.cancel() break elif gen_product_data['status'] == "SUCCESS": diff --git a/app/service/generate_image/service_generate_relight_image.py b/app/service/generate_image/service_generate_relight_image.py index 3efbe88..2e48ae2 100644 --- a/app/service/generate_image/service_generate_relight_image.py +++ b/app/service/generate_image/service_generate_relight_image.py @@ -121,7 +121,7 @@ class GenerateRelightImage: time_out = 600 while time_out > 0: gen_product_data, _ = self.read_tasks_status() - if gen_product_data['status'] in ["REVOKED", "FAILURE"]: + if gen_product_data['status'] in ["REVOKED", "FAILURE", "NO_FACE"]: ctx.cancel() break elif gen_product_data['status'] == "SUCCESS":