From 3eef816acd50fdfad1e9ec517250e08db84972bf Mon Sep 17 00:00:00 2001
From: sirius422 <38351200+sirius422@users.noreply.github.com>
Date: Tue, 16 Jul 2024 02:07:27 +0800
Subject: [PATCH] Update wd14tagger.py download_model to use proxy

Read proxy settings from environment variables if available by default
---
 wd14tagger.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wd14tagger.py b/wd14tagger.py
index c88f583..7dfdc7a 100644
--- a/wd14tagger.py
+++ b/wd14tagger.py
@@ -117,7 +117,7 @@ async def download_model(model, client_id, node):
     url = config["models"][model]
     url = url.replace("{HF_ENDPOINT}", hf_endpoint)
     url = f"{url}/resolve/main/"
-    async with aiohttp.ClientSession(loop=asyncio.get_event_loop()) as session:
+    async with aiohttp.ClientSession(loop=asyncio.get_event_loop(), trust_env=True) as session:
         async def update_callback(perc):
             nonlocal client_id
             message = ""