wasmdashai commited on
Commit
7f086ae
·
verified ·
1 Parent(s): a90af58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -173,8 +173,8 @@ def get_data_loader(train_dataset_dirs,eval_dataset_dir,full_generation_dir,dev
173
 
174
 
175
  eval_dataset = None
176
- if training_args.do_eval:
177
- eval_dataset = FeaturesCollectionDataset(dataset_dir = eval_dataset_dir,
178
  device = device
179
  )
180
 
@@ -561,10 +561,7 @@ train_dataset_dirs=[
561
 
562
  dir_model='wasmdashai/vits-ar-huba-fine'
563
 
564
- ctrain_datasets,eval_dataset,full_generation_dataset=get_data_loader(train_dataset_dirs = train_dataset_dirs,
565
- eval_dataset_dir = os.path.join(dataset_dir,'eval'),
566
- full_generation_dir = os.path.join(dataset_dir,'full_generation'),
567
- device=device)
568
 
569
  global_step=0
570
  wandb.login(key= "782b6a6e82bbb5a5348de0d3c7d40d1e76351e79")
@@ -594,7 +591,10 @@ def greet(text,id):
594
  print(device)
595
 
596
  b=int(id)
597
-
 
 
 
598
 
599
  print('load Data')
600
  wandb.init(project= 'AZ')
 
173
 
174
 
175
  eval_dataset = None
176
+
177
+ eval_dataset = FeaturesCollectionDataset(dataset_dir = eval_dataset_dir,
178
  device = device
179
  )
180
 
 
561
 
562
  dir_model='wasmdashai/vits-ar-huba-fine'
563
 
564
+
 
 
 
565
 
566
  global_step=0
567
  wandb.login(key= "782b6a6e82bbb5a5348de0d3c7d40d1e76351e79")
 
591
  print(device)
592
 
593
  b=int(id)
594
+ ctrain_datasets,eval_dataset,full_generation_dataset=get_data_loader(train_dataset_dirs = train_dataset_dirs,
595
+ eval_dataset_dir = os.path.join(dataset_dir,'eval'),
596
+ full_generation_dir = os.path.join(dataset_dir,'full_generation'),
597
+ device=device)
598
 
599
  print('load Data')
600
  wandb.init(project= 'AZ')