Update app.py

app.py

@@ -9,9 +9,7 @@ from VitsModelSplit.vits_model2 import VitsModel,get_state_grad_loss
 import VitsModelSplit.monotonic_align  as  monotonic_align
 
 token=os.environ.get("key_")
-tokenizer = AutoTokenizer.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)
-#device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model_vits=VitsModel.from_pretrained("wasmdashai/vits-ar-sa-huba",token=token)#.to(device)
+
 
 # import VitsModelSplit.monotonic_align as monotonic_align
 from IPython.display import clear_output
@@ -188,7 +186,7 @@ def  train_step(batch,models=[],optimizers=[], training_args=None,tools=[]):
      feature_extractor,maf,dict_state_grad_loss=tools
      
      with autocast(enabled=training_args.fp16):
-          speaker_embeddings=get_embed_speaker(model,batch["speaker_id"])
+          speaker_embeddings=get_embed_speaker(self,batch["speaker_id"])
           waveform,ids_slice,log_duration,prior_latents,posterior_log_variances,prior_means,prior_log_variances,labels_padding_mask = self.forward_train(
               input_ids=batch["input_ids"],
               attention_mask=batch["attention_mask"],