Update app.py

app.py

@@ -561,8 +561,10 @@ train_dataset_dirs=[
 
 dir_model='wasmdashai/vits-ar-huba-fine'
 
-
-
+ctrain_datasets,eval_dataset,full_generation_dataset=get_data_loader(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'),
+                                                                     device=device)
 
 global_step=0
 wandb.login(key= "782b6a6e82bbb5a5348de0d3c7d40d1e76351e79")
@@ -593,10 +595,7 @@ def greet(text,id):
     
     b=int(id)
     
-    ctrain_datasets,eval_dataset,full_generation_dataset=get_data_loader(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'),
-                                                                     device=device)
+
     print('load Data')
     wandb.init(project= 'AZ')
     print('wandb')