From 1ceb4477d490f1caa47af507391b17a45f631aa4 Mon Sep 17 00:00:00 2001
From: VSlobolinskyi <vitalii.slobolinskyi@gmail.com>
Date: Tue, 18 Mar 2025 01:31:03 +0200
Subject: [PATCH] Add --share argument

---
 configs/config.py        | 5 +++++
 infer-web.py             | 2 +-
 pyproject.toml           | 2 +-
 tools/download_assets.py | 2 +-
 4 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/configs/config.py b/configs/config.py
index a330fb5..9b1028c 100644
--- a/configs/config.py
+++ b/configs/config.py
@@ -78,6 +78,11 @@ class Config:
         exe = sys.executable or "python"
         parser = argparse.ArgumentParser()
         parser.add_argument("--port", type=int, default=7865, help="Listen port")
+        parser.add_argument(
+            "--share",
+            action="store_true",
+            help="If set, create a shareable public link."
+        )
         parser.add_argument("--pycmd", type=str, default=exe, help="Python command")
         parser.add_argument("--colab", action="store_true", help="Launch in colab")
         parser.add_argument(
diff --git a/infer-web.py b/infer-web.py
index 47596d5..aaf0de6 100644
--- a/infer-web.py
+++ b/infer-web.py
@@ -1612,7 +1612,7 @@ with gr.Blocks(title="RVC WebUI") as app:
         app.queue(concurrency_count=511, max_size=1022).launch(share=True)
     else:
         app.queue(concurrency_count=511, max_size=1022).launch(
-            server_name="0.0.0.0",
+            server_name="localhost",
             inbrowser=not config.noautoopen,
             server_port=config.listen_port,
             quiet=True,
diff --git a/pyproject.toml b/pyproject.toml
index b64d736..27d7502 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -57,7 +57,7 @@ fastapi = "0.88"
 torchfcpe = "^0.0.4"
 ffmpy = "0.3.1"
 torchvision = "0.19.0"
-[tool.poetry.dev-dependencies]
+[tool.poetry.group.dev.dependencies]
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
diff --git a/tools/download_assets.py b/tools/download_assets.py
index 63f0220..b7ca3a8 100644
--- a/tools/download_assets.py
+++ b/tools/download_assets.py
@@ -21,7 +21,7 @@ if __name__ == "__main__":
     dl_model(RVC_DOWNLOAD_LINK, "hubert_base.pt", BASE_DIR / "assets/hubert")
     print("Downloading rmvpe.pt...")
     dl_model(RVC_DOWNLOAD_LINK, "rmvpe.pt", BASE_DIR / "assets/rmvpe")
-    print("Downloading rmvpe.pt...")
+    print("Downloading rmvpe.onnx...")
     dl_model(RVC_DOWNLOAD_LINK, "rmvpe.onnx", BASE_DIR / "assets/rmvpe")
     print("Downloading vocals.onnx...")
     dl_model(