Update app.py

app.py

@@ -360,7 +360,7 @@ with gr.Blocks() as demo:
         else:
             cost_time = 0
         status, video_url = get_result_with_task_id(task_id)
-        if task_status not in task_status:
+        if task_id not in task_status:
             task_status[task_id] = {"status": status, "url": video_url, "time": time.time()}
         else:
             task_status[task_id]["status"] = status