Update app.py

app.py

@@ -585,14 +585,7 @@ train_dataset_dirs=[
 
 
 dir_model='wasmdashai/vits-ar-huba-fine'
-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
-                    )
+pro=None
 
 @spaces.GPU
 def run_train_epoch(num):
@@ -607,7 +600,15 @@ def init_training():
 
 @spaces.GPU
 def init_Starting():
-    pro.init_Starting()
+    globals pro
+    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():