Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
177 |
-
|
178 |
device = device
|
179 |
)
|
180 |
|
@@ -561,10 +561,7 @@ train_dataset_dirs=[
|
|
561 |
|
562 |
dir_model='wasmdashai/vits-ar-huba-fine'
|
563 |
|
564 |
-
|
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')
|