chaojiemao commited on
Commit
c76f952
·
verified ·
1 Parent(s): 14a42e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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 = 2
19
- LIMIT_HISTORY_RUNING_TASK = 4
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(), gr.Slider(), gr.Slider()
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), 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,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