chaojiemao commited on
Commit
e67efa8
·
verified ·
1 Parent(s): 8dcfa93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -361,13 +361,14 @@ with gr.Blocks() as demo:
361
  cost_time = 0
362
  status, video_url = get_result_with_task_id(task_id)
363
  if task_id not in task_status:
364
- task_status[task_id] = {"status": status, "url": video_url, "time": time.time()}
365
  else:
366
  task_status[task_id]["status"] = status
367
  task_status[task_id]["url"] = video_url
368
  waiting_time = get_waiting_time(task_id)
369
  value = task_status.get(task_id, {"value": 100})["value"]
370
  value = min(value, int(cost_time*100/waiting_time))
 
371
  if not video_url == "" and status: value = 100
372
  process_bar = gr.Slider(label=f"({value}%)Generating" if value % 2 == 1 else f"({value}%)Generating.....", value=value)
373
  process_change_ret = process_change(task_id, task)
 
361
  cost_time = 0
362
  status, video_url = get_result_with_task_id(task_id)
363
  if task_id not in task_status:
364
+ task_status[task_id] = {"status": status, "url": video_url, "time": time.time(), "value": 0 if status else 100}
365
  else:
366
  task_status[task_id]["status"] = status
367
  task_status[task_id]["url"] = video_url
368
  waiting_time = get_waiting_time(task_id)
369
  value = task_status.get(task_id, {"value": 100})["value"]
370
  value = min(value, int(cost_time*100/waiting_time))
371
+ task_status[task_id]["value"] = value
372
  if not video_url == "" and status: value = 100
373
  process_bar = gr.Slider(label=f"({value}%)Generating" if value % 2 == 1 else f"({value}%)Generating.....", value=value)
374
  process_change_ret = process_change(task_id, task)