From 6a3e1ca94cabdbb178db4691b392b3046574f66b Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Sat, 30 Dec 2023 09:16:05 +0900
Subject: [PATCH] Feat: in colab, use Google drive path as dataset_path

---
 webui_train.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/webui_train.py b/webui_train.py
index 41486f3..2b749b1 100644
--- a/webui_train.py
+++ b/webui_train.py
@@ -11,8 +11,15 @@ from tools.log import logger
 from tools.subprocess_utils import run_script_with_log, second_elem_of
 
 
+is_colab = "google.colab" in sys.modules
+
+
 def get_path(model_name):
     assert model_name != "", "モデル名は空にできません"
+    if is_colab:
+        dataset_path = os.path.join(
+            "/content/drive/MyDrive/Style-Bert-VITS2/Data", model_name
+        )
     dataset_path = os.path.join("Data", model_name)
     lbl_path = os.path.join(dataset_path, "esd.list")
     train_path = os.path.join(dataset_path, "train.list")