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

Merge branch 'refs/heads/develop'

Browse Source
This commit is contained in:
zhouchengrong
2025-02-13 13:05:12 +08:00
parent d26f38359c 9ba883a6ee
commit 516bbf40e4
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/service/generate_image/service_generate_product_image.py
View File

@@ -302,7 +302,7 @@ class GenerateProductImage:
time_out = 600 time_out = 600
while time_out > 0: while time_out > 0:
gen_product_data, _ = self.read_tasks_status() 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() ctx.cancel()
break break
elif gen_product_data['status'] == "SUCCESS": elif gen_product_data['status'] == "SUCCESS":

2
app/service/generate_image/service_generate_relight_image.py
View File

@@ -121,7 +121,7 @@ class GenerateRelightImage:
time_out = 600 time_out = 600
while time_out > 0: while time_out > 0:
gen_product_data, _ = self.read_tasks_status() 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() ctx.cancel()
break break
elif gen_product_data['status'] == "SUCCESS": elif gen_product_data['status'] == "SUCCESS":