chaojiemao commited on
Commit
61e74bc
·
verified ·
1 Parent(s): 49bc9df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -312,7 +312,7 @@ with gr.Blocks() as demo:
312
 
313
 
314
  def process_change(task_id, task):
315
- status = task_status[task_id]["status"]
316
  if status:
317
  video_url = task_status[task_id]["url"]
318
  ret_t2v_btn = gr.Button(visible=True) if task == 't2v' else gr.Button()
 
312
 
313
 
314
  def process_change(task_id, task):
315
+ status = task_status.get(task_id, {"status":False})["status"]
316
  if status:
317
  video_url = task_status[task_id]["url"]
318
  ret_t2v_btn = gr.Button(visible=True) if task == 't2v' else gr.Button()