wasmdashai commited on
Commit
2835be9
·
verified ·
1 Parent(s): face690

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -9
app.py CHANGED
@@ -585,14 +585,7 @@ train_dataset_dirs=[
585
 
586
 
587
  dir_model='wasmdashai/vits-ar-huba-fine'
588
- pro=TrinerModelVITS(dir_model=dir_model,
589
- path_training_args='VitsModelSplit/finetune_config_ara.json',
590
- train_dataset_dirs = train_dataset_dirs,
591
- eval_dataset_dir = os.path.join(dataset_dir,'eval'),
592
- full_generation_dir = os.path.join(dataset_dir,'full_generation'),
593
- token=token,
594
- device=device
595
- )
596
 
597
  @spaces.GPU
598
  def run_train_epoch(num):
@@ -607,7 +600,15 @@ def init_training():
607
 
608
  @spaces.GPU
609
  def init_Starting():
610
- pro.init_Starting()
 
 
 
 
 
 
 
 
611
  return 'init_Starting'
612
  @spaces.GPU
613
  def init_wandb():
 
585
 
586
 
587
  dir_model='wasmdashai/vits-ar-huba-fine'
588
+ pro=None
 
 
 
 
 
 
 
589
 
590
  @spaces.GPU
591
  def run_train_epoch(num):
 
600
 
601
  @spaces.GPU
602
  def init_Starting():
603
+ globals pro
604
+ pro=TrinerModelVITS(dir_model=dir_model,
605
+ path_training_args='VitsModelSplit/finetune_config_ara.json',
606
+ train_dataset_dirs = train_dataset_dirs,
607
+ eval_dataset_dir = os.path.join(dataset_dir,'eval'),
608
+ full_generation_dir = os.path.join(dataset_dir,'full_generation'),
609
+ token=token,
610
+ device=device
611
+ )
612
  return 'init_Starting'
613
  @spaces.GPU
614
  def init_wandb():