Spaces:
Sleeping
Sleeping
Create main.py
Browse files
main.py
ADDED
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# main.py
|
2 |
+
|
3 |
+
import os
|
4 |
+
from gui import get_gui
|
5 |
+
from utils import download_manager
|
6 |
+
|
7 |
+
MDX_DOWNLOAD_LINK = "https://github.com/TRvlvr/model_repo/releases/download/all_public_uvr_models/"
|
8 |
+
UVR_MODELS = [
|
9 |
+
"UVR-MDX-NET-Voc_FT.onnx",
|
10 |
+
"UVR_MDXNET_KARA_2.onnx",
|
11 |
+
"Reverb_HQ_By_FoxJoy.onnx",
|
12 |
+
"UVR-MDX-NET-Inst_HQ_4.onnx",
|
13 |
+
]
|
14 |
+
|
15 |
+
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
16 |
+
mdxnet_models_dir = os.path.join(BASE_DIR, "mdx_models")
|
17 |
+
os.makedirs(mdxnet_models_dir, exist_ok=True)
|
18 |
+
|
19 |
+
|
20 |
+
def download_models():
|
21 |
+
for model in UVR_MODELS:
|
22 |
+
url = os.path.join(MDX_DOWNLOAD_LINK, model)
|
23 |
+
download_manager(url, mdxnet_models_dir)
|
24 |
+
|
25 |
+
|
26 |
+
if __name__ == "__main__":
|
27 |
+
download_models()
|
28 |
+
theme = "NoCrypt/miku"
|
29 |
+
app = get_gui(theme)
|
30 |
+
app.queue(default_concurrency_limit=40)
|
31 |
+
app.launch(
|
32 |
+
max_threads=40,
|
33 |
+
share=False,
|
34 |
+
show_error=True,
|
35 |
+
quiet=False,
|
36 |
+
debug=False,
|
37 |
+
)
|