Update app.py

app.py

@@ -127,7 +127,7 @@ def get_state_grad_loss(k1=True,
                              discriminator=True):
                              return {'k1':k1,'mel':mel,'duration':duration,'generator':generator,'discriminator':discriminator}
 
-
+@spaces.GPU
 def clip_grad_value_(parameters, clip_value, norm_type=2):
   if isinstance(parameters, torch.Tensor):
     parameters = [parameters]
@@ -145,7 +145,7 @@ def clip_grad_value_(parameters, clip_value, norm_type=2):
   total_norm = total_norm ** (1. / norm_type)
   return total_norm
 
-
+@spaces.GPU
 def get_embed_speaker(self,speaker_id):
      if self.config.num_speakers > 1 and speaker_id is not None:
           if isinstance(speaker_id, int):
@@ -160,6 +160,7 @@ def get_embed_speaker(self,speaker_id):
           return self.embed_speaker(speaker_id).unsqueeze(-1)
      else:
           return None
+@spaces.GPU
 def  get_data_loader(train_dataset_dirs,eval_dataset_dir,full_generation_dir,device):
      ctrain_datasets=[]
      for  dataset_dir ,id_sp in train_dataset_dirs: