Spaces:
Sleeping
Sleeping
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')
|