From 6136f4062aeb2dacbaaf63d02a474278915a5c25 Mon Sep 17 00:00:00 2001
From: magic-akari <akari.ccino@gmail.com>
Date: Sat, 15 Jul 2023 17:02:03 +0800
Subject: [PATCH] fix(colab): add feature retrieval

---
 sovits4_for_colab.ipynb | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/sovits4_for_colab.ipynb b/sovits4_for_colab.ipynb
index 2319848..c5f4966 100644
--- a/sovits4_for_colab.ipynb
+++ b/sovits4_for_colab.ipynb
@@ -453,7 +453,25 @@
     "#@markdown\n",
     "\n",
     "%cd /content/so-vits-svc\n",
-    "!python cluster/train_cluster.py"
+    "!python cluster/train_cluster.py --gpu"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#@title Train index model (Optional)\n",
+    "\n",
+    "#@markdown # Train index model (Optional)\n",
+    "\n",
+    "#@markdown #### Details see [README.md#feature-retrieval](https://github.com/svc-develop-team/so-vits-svc#feature-retrieval)\n",
+    "\n",
+    "#@markdown\n",
+    "\n",
+    "%cd /content/so-vits-svc\n",
+    "!python train_index.py -c configs/config.json"
    ]
   },
   {
@@ -501,14 +519,15 @@
     "```JavaScript\n",
     "const ping = () => {\n",
     "  const btn = document.querySelector(\"colab-connect-button\");\n",
-    "  if (btn) {\n",
-    "    btn.click();\n",
+    "  const inner_btn = btn.shadowRoot.querySelector(\"#connect\");\n",
+    "  if (inner_btn) {\n",
+    "    inner_btn.click();\n",
     "    console.log(\"Clicked on connect button\");\n",
     "  } else {\n",
     "    console.log(\"connect button not found\");\n",
     "  }\n",
     "\n",
-    "  const nextTime = 5000 + Math.random() * 1000;\n",
+    "  const nextTime = 50000 + Math.random() * 10000;\n",
     "\n",
     "  setTimeout(ping, nextTime);\n",
     "};\n",