Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -433,8 +433,8 @@ class TrinerModelVITS:
|
|
433 |
|
434 |
def load_dataset(self,train_dataset_dirs,eval_dataset_dir,full_generation_dir):
|
435 |
ctrain_datasets,eval_dataset,full_generation_dataset=get_data_loader(train_dataset_dirs = train_dataset_dirs,
|
436 |
-
eval_dataset_dir =
|
437 |
-
full_generation_dir =
|
438 |
device=self.device)
|
439 |
self.DataSets={'train':ctrain_datasets,'eval':eval_dataset,'full_generation':full_generation_dataset}
|
440 |
|
|
|
433 |
|
434 |
def load_dataset(self,train_dataset_dirs,eval_dataset_dir,full_generation_dir):
|
435 |
ctrain_datasets,eval_dataset,full_generation_dataset=get_data_loader(train_dataset_dirs = train_dataset_dirs,
|
436 |
+
eval_dataset_dir =eval_dataset_dir ,
|
437 |
+
full_generation_dir =full_generation_dir ,
|
438 |
device=self.device)
|
439 |
self.DataSets={'train':ctrain_datasets,'eval':eval_dataset,'full_generation':full_generation_dataset}
|
440 |
|