Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -13,9 +13,9 @@ DASHSCOPE_API_KEY = os.getenv('DASHSCOPE_API_KEY')
|
|
13 |
dashscope.api_key = DASHSCOPE_API_KEY
|
14 |
|
15 |
KEEP_SUCCESS_TASK = 3600 * 10
|
16 |
-
KEEP_RUNING_TASK = 3600 *
|
17 |
# the total running task number in 1800 seconds
|
18 |
-
LIMIT_RUNING_TASK =
|
19 |
|
20 |
def t2v_generation(prompt, resolution, watermark_wan, seed = -1):
|
21 |
seed = seed if seed >= 0 else random.randint(0, 2147483647)
|
@@ -102,10 +102,13 @@ task_status = {}
|
|
102 |
|
103 |
def allow_task_num():
|
104 |
num = 0
|
|
|
105 |
for task_id in task_status:
|
106 |
if not task_status[task_id]["status"] and task_status[task_id]["time"] + 1800 > time.time():
|
107 |
num += 1
|
108 |
-
|
|
|
|
|
109 |
def clean_task_status():
|
110 |
# clean the task over 1800 seconds
|
111 |
for task_id in copy.deepcopy(task_status):
|
@@ -125,6 +128,10 @@ def cost_time(task_id):
|
|
125 |
return f"{et:.2f}"
|
126 |
else:
|
127 |
return gr.Textbox()
|
|
|
|
|
|
|
|
|
128 |
def get_process_bar(task_id, status):
|
129 |
clean_task_status()
|
130 |
if task_id not in task_status:
|
|
|
13 |
dashscope.api_key = DASHSCOPE_API_KEY
|
14 |
|
15 |
KEEP_SUCCESS_TASK = 3600 * 10
|
16 |
+
KEEP_RUNING_TASK = 3600 * 1
|
17 |
# the total running task number in 1800 seconds
|
18 |
+
LIMIT_RUNING_TASK = 4
|
19 |
|
20 |
def t2v_generation(prompt, resolution, watermark_wan, seed = -1):
|
21 |
seed = seed if seed >= 0 else random.randint(0, 2147483647)
|
|
|
102 |
|
103 |
def allow_task_num():
|
104 |
num = 0
|
105 |
+
total_num = 0
|
106 |
for task_id in task_status:
|
107 |
if not task_status[task_id]["status"] and task_status[task_id]["time"] + 1800 > time.time():
|
108 |
num += 1
|
109 |
+
if not task_status[task_id]["status"]:
|
110 |
+
total_num += 1
|
111 |
+
return num < LIMIT_RUNING_TASK or num < 10
|
112 |
def clean_task_status():
|
113 |
# clean the task over 1800 seconds
|
114 |
for task_id in copy.deepcopy(task_status):
|
|
|
128 |
return f"{et:.2f}"
|
129 |
else:
|
130 |
return gr.Textbox()
|
131 |
+
|
132 |
+
def waiting_time():
|
133 |
+
|
134 |
+
|
135 |
def get_process_bar(task_id, status):
|
136 |
clean_task_status()
|
137 |
if task_id not in task_status:
|