Update app.py

app.py

@@ -407,9 +407,9 @@ class TrinerModelVITS:
         self.epoch_count=0
         self.global_step=0
         self.len_dataset=len(self.DataSets['train'])
-        self.init_Starting()
+        self.load_model()
         self.init_wandb()
-        self.init_training()
+        
        
         
 
@@ -422,7 +422,7 @@ class TrinerModelVITS:
         
     def init_training(self):
         
-        self.load_model()
+        
         self.initialize_training_components()
         self.epoch_count=0
         
@@ -585,10 +585,18 @@ train_dataset_dirs=[
 
 
 dir_model='wasmdashai/vits-ar-huba-fine'
-pro=None
+pro=TrinerModelVITS(dir_model=dir_model,
+                    path_training_args='VitsModelSplit/finetune_config_ara.json',
+                    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
+                    )
 
 @spaces.GPU
 def run_train_epoch(num):
+   pro.init_training()
    for i in range(10):
     # model.train(True)
         yield pro.run_train_epoch()
@@ -600,17 +608,10 @@ def init_training():
 
 @spaces.GPU
 def init_Starting():
-    global pro
+     
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-    pro=TrinerModelVITS(dir_model=dir_model,
-                    path_training_args='VitsModelSplit/finetune_config_ara.json',
-                    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
-                    )
+    
     return 'init_Starting'
 @spaces.GPU
 def init_wandb():