diff --git a/nodes/llm_nodes.py b/nodes/llm_nodes.py index 07eca21..1a59dd5 100644 --- a/nodes/llm_nodes.py +++ b/nodes/llm_nodes.py @@ -329,11 +329,13 @@ class ModalClothesMask: job_id = job_resp["taskId"] wait_time = 240 - interval = 3 + interval = 2 logger.info("开始轮询任务状态") + sleep(1) for _ in range(0, wait_time, interval): logger.info("查询任务状态") - result = send_request("get", f"https://{endpoint}/google/{job_id}", timeout=timeout) + result = send_request("get", f"https://{endpoint}/google/{job_id}", + headers={'Authorization': 'Bearer bowong7777'}, timeout=timeout) if result.status_code == 200: result = result.json() if result["status"] == "success": @@ -355,7 +357,7 @@ class ModalEditCustom: return { "required": { "image": ("IMAGE",), - "prompt": ("STRING", {"default": "将背景去除,输出原尺寸图片","multiline": True}), + "prompt": ("STRING", {"default": "将背景去除,输出原尺寸图片", "multiline": True}), "endpoint": ("STRING", {"default": "bowongai-dev--bowong-ai-video-gemini-fastapi-webapp.modal.run"}), }, } @@ -391,11 +393,13 @@ class ModalEditCustom: job_id = job_resp["taskId"] wait_time = 240 - interval = 3 + interval = 2 logger.info("开始轮询任务状态") + sleep(1) for _ in range(0, wait_time, interval): logger.info("查询任务状态") - result = send_request("get", f"https://{endpoint}/google/{job_id}", timeout=timeout) + result = send_request("get", f"https://{endpoint}/google/{job_id}", + headers={'Authorization': 'Bearer bowong7777'}, timeout=timeout) if result.status_code == 200: result = result.json() if result["status"] == "success":