Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -15,8 +15,8 @@ dashscope.api_key = DASHSCOPE_API_KEY
|
|
| 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 =
|
| 19 |
-
LIMIT_HISTORY_RUNING_TASK =
|
| 20 |
|
| 21 |
task_status = {}
|
| 22 |
|
|
@@ -73,7 +73,7 @@ def t2v_generation_async(prompt, size, watermark_wan, seed=-1):
|
|
| 73 |
return task_id, status, gr.Button(visible=False), gr.Button(visible=True), get_cost_time(task_id), get_waiting_time(task_id)
|
| 74 |
except Exception as e:
|
| 75 |
gr.Warning(f"Warning: {e}")
|
| 76 |
-
return None, True, gr.Button(), gr.Button(),
|
| 77 |
|
| 78 |
|
| 79 |
def i2v_generation(prompt, image, watermark_wan, seed=-1):
|
|
@@ -103,7 +103,7 @@ def i2v_generation_async(prompt, image, watermark_wan, seed=-1):
|
|
| 103 |
print(rsp)
|
| 104 |
task_id = rsp.output.task_id
|
| 105 |
status = False
|
| 106 |
-
return task_id, status, gr.Button(visible=False), gr.Button(visible=True),
|
| 107 |
except Exception as e:
|
| 108 |
gr.Warning(f"Warning: {e}")
|
| 109 |
return "", None, gr.Button(), gr.Button(), gr.Slider(), gr.Slider()
|
|
@@ -355,7 +355,7 @@ with gr.Blocks() as demo:
|
|
| 355 |
waiting_time = get_waiting_time(task_id)
|
| 356 |
value = task_status.get(task_id, {"value": 100})["value"]
|
| 357 |
value = min(value, int(cost_time*100/waiting_time))
|
| 358 |
-
if not video_url == "": value = 100
|
| 359 |
process_bar = gr.Slider(label=f"({value}%)Generating" if value % 2 == 1 else f"({value}%)Generating.....", value=value)
|
| 360 |
process_change_ret = process_change(task_id, task)
|
| 361 |
return *process_change_ret, cost_time, waiting_time, process_bar
|
|
|
|
| 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 = 20
|
| 19 |
+
LIMIT_HISTORY_RUNING_TASK = 40
|
| 20 |
|
| 21 |
task_status = {}
|
| 22 |
|
|
|
|
| 73 |
return task_id, status, gr.Button(visible=False), gr.Button(visible=True), get_cost_time(task_id), get_waiting_time(task_id)
|
| 74 |
except Exception as e:
|
| 75 |
gr.Warning(f"Warning: {e}")
|
| 76 |
+
return None, True, gr.Button(), gr.Button(), gr.Slider(), gr.Slider()
|
| 77 |
|
| 78 |
|
| 79 |
def i2v_generation(prompt, image, watermark_wan, seed=-1):
|
|
|
|
| 103 |
print(rsp)
|
| 104 |
task_id = rsp.output.task_id
|
| 105 |
status = False
|
| 106 |
+
return task_id, status, gr.Button(visible=False), gr.Button(visible=True), get_cost_time(task_id), get_waiting_time(task_id)
|
| 107 |
except Exception as e:
|
| 108 |
gr.Warning(f"Warning: {e}")
|
| 109 |
return "", None, gr.Button(), gr.Button(), gr.Slider(), gr.Slider()
|
|
|
|
| 355 |
waiting_time = get_waiting_time(task_id)
|
| 356 |
value = task_status.get(task_id, {"value": 100})["value"]
|
| 357 |
value = min(value, int(cost_time*100/waiting_time))
|
| 358 |
+
if not video_url == "" and status: value = 100
|
| 359 |
process_bar = gr.Slider(label=f"({value}%)Generating" if value % 2 == 1 else f"({value}%)Generating.....", value=value)
|
| 360 |
process_change_ret = process_change(task_id, task)
|
| 361 |
return *process_change_ret, cost_time, waiting_time, process_bar
|