Update app.py

app.py

@@ -406,6 +406,7 @@ class TrinerModelVITS:
         self.load_dataset(train_dataset_dirs,eval_dataset_dir,full_generation_dir)
         self.epoch_count=0
         self.global_step=0
+        self.len_dataset=len(self.DataSets['train'])
        
         
 
@@ -415,7 +416,7 @@ class TrinerModelVITS:
         self.stute_mode=False
         print('end training_args')
         
-        self.len_dataset=len(self.DataSets['train'])
+        
     def init_training(self):
         
         self.load_model()