show error if dataset is empty

ddPn08 · ddPn08 · commit 8f556db7925c · 2023-05-22T19:35:49.000+09:00
diff --git a/modules/tabs/training.py b/modules/tabs/training.py
@@ -61,6 +61,9 @@ def train_index_only(
                 recursive=recursive,
             )
 
+            if len(datasets) == 0:
+                raise Exception("No audio files found")
+
             yield "Preprocessing..."
             split.preprocess_audio(
                 datasets,
@@ -159,6 +162,9 @@ def train_all(
                 recursive=recursive,
             )
 
+            if len(datasets) == 0:
+                raise Exception("No audio files found")
+
             yield "Preprocessing..."
             split.preprocess_audio(
                 datasets,
