Update app.py

app.py

@@ -181,6 +181,7 @@ def  get_data_loader(train_dataset_dirs,eval_dataset_dir,full_generation_dir,dev
                                                         device = device)
      return ctrain_datasets,eval_dataset,full_generation_dataset
 global_step=0
+@spaces.GPU
 def trainer_to_cuda(self,
           ctrain_datasets = None,
           eval_dataset = None,
@@ -563,7 +564,7 @@ dir_model='wasmdashai/vits-ar-huba-fine'
 
 
 global_step=0
-
+wandb.login(key= "782b6a6e82bbb5a5348de0d3c7d40d1e76351e79")
 @spaces.GPU
 def greet(text,id):
     global GK 
@@ -595,9 +596,9 @@ def greet(text,id):
                                                               eval_dataset_dir = os.path.join(dataset_dir,'eval'),
                                                             full_generation_dir = os.path.join(dataset_dir,'full_generation'),
                                                                      device=device)
-    wandb.login(key= "782b6a6e82bbb5a5348de0d3c7d40d1e76351e79")
+    
     wandb.init(project= 'AZ',config = training_args.to_dict())
-
+    print('wandb')
     for i in range(10000):
 # model.train(True)
         print(f'clcye epochs ={i}')
@@ -605,7 +606,7 @@ def greet(text,id):
         model=VitsModel.from_pretrained(dir_model,token=token).to(device)
         # model.setMfA(monotonic_align.maximum_path)
         #dir_model_save=dir_model+'/vend'
-    
+        print('loadeed')
     
         trainer_to_cuda(model,
                         ctrain_datasets = ctrain_datasets,