wasmdashai commited on
Commit
3232fb7
·
verified ·
1 Parent(s): 7d0e0a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = os.path.join(dataset_dir,'eval'),
437
- full_generation_dir = os.path.join(dataset_dir,'full_generation'),
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