Update app.py

app.py

@@ -403,7 +403,7 @@ class TrinerModelVITS:
         self.stute_mode=False
         self.token=token
     
-       
+        self.load_dataset(train_dataset_dirs,eval_dataset_dir,full_generation_dir)
         self.epoch_count=0
         self.global_step=0
        
@@ -413,7 +413,7 @@ class TrinerModelVITS:
         self.training_args=load_training_args(self.path_training_args)
         self.stute_mode=False
        
-        self.load_dataset(train_dataset_dirs,eval_dataset_dir,full_generation_dir)
+        
         self.len_dataset=len(self.DataSets['train'])
     def init_training(self):
         
@@ -577,6 +577,7 @@ pro=TrinerModelVITS(dir_model=dir_model,
                     train_dataset_dirs = train_dataset_dirs,
                                                               eval_dataset_dir = os.path.join(dataset_dir,'eval'),
                                                             full_generation_dir = os.path.join(dataset_dir,'full_generation'),
+                                                            token=token,
                                                                      device=device
                     )