From 380fd52085a27118692ecb567c2f2cd485d381bb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Mon, 22 Jan 2024 22:28:09 +0900
Subject: [PATCH] optimize(infer-web): silent httpx INFO logging

and add rmvpe downloading in dlmodels.bat
---
 infer-web.py       |  1 +
 tools/dlmodels.bat | 16 +++++++++++++++-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/infer-web.py b/infer-web.py
index 99b5494..9a129b6 100644
--- a/infer-web.py
+++ b/infer-web.py
@@ -34,6 +34,7 @@ import logging
 
 
 logging.getLogger("numba").setLevel(logging.WARNING)
+logging.getLogger("httpx").setLevel(logging.WARNING)
 
 logger = logging.getLogger(__name__)
 
diff --git a/tools/dlmodels.bat b/tools/dlmodels.bat
index 5d80f50..b83825a 100644
--- a/tools/dlmodels.bat
+++ b/tools/dlmodels.bat
@@ -63,6 +63,9 @@ set hb=hubert_base.pt
 
 set dlhb=https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/hubert_base.pt
 
+set rmvpe=rmvpe.pt
+set dlrmvpe=https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/rmvpe.pt
+
 echo dir check start.
 echo=
 
@@ -332,7 +335,7 @@ if exist "%~dp0assets\uvr5_weights\onnx_dereverb_By_FoxJoy\%onnx_dereverb%" (
 
 echo checking %hb%
 if exist "%~dp0assets\hubert\%hb%" (
-        echo %hb% in .\assets\hubert\pretrained checked.
+        echo %hb% in .\assets\hubert checked.
         echo=
     ) else (
         echo failed. starting download from huggingface.
@@ -341,6 +344,17 @@ if exist "%~dp0assets\hubert\%hb%" (
         echo=)
     )
 
+echo checking %rmvpe%
+if exist "%~dp0assets\rmvpe\%rmvpe%" (
+        echo %rmvpe% in .\assets\rmvpe checked.
+        echo=
+    ) else (
+        echo failed. starting download from huggingface.
+        %~dp0%aria2%\aria2c --console-log-level=error -c -x 16 -s 16 -k 1M %dlrmvpe% -d %~dp0assets\rmvpe\ -o %rmvpe%
+        if exist "%~dp0assets\rmvpe\%rmvpe%" (echo download successful.) else (echo please try again!
+        echo=)
+    )
+
 echo required files check finished.
 echo envfiles check complete.
 pause