diff --git a/tgbot/shit/prompting.py b/tgbot/shit/prompting.py index c426738..f30b62c 100644 --- a/tgbot/shit/prompting.py +++ b/tgbot/shit/prompting.py @@ -26,7 +26,7 @@ scheduled_prompts: dict = {} async def ws_task(): - ws_url = f"wss://{IMAGE_GENERATOR_API_HOST}/ws?token={IMAGE_GENERATOR_TOKEN}" + ws_url = f"wss://{IMAGE_GENERATOR_API_HOST}/system/ws?token={IMAGE_GENERATOR_TOKEN}" retry_delay = 1 # Initial delay (1s), will increase with backoff while True: @@ -74,7 +74,7 @@ thread.start() async def enqueue_prompt_t2i(args: dict): - url = f"https://{IMAGE_GENERATOR_API_HOST}/prompt/generic/txt2img/" + url = f"https://{IMAGE_GENERATOR_API_HOST}/prompting/txt2img/" async with aiohttp.ClientSession() as session: headers = {"Authorization": "Bearer " + IMAGE_GENERATOR_TOKEN