Update app.py

app.py

@@ -477,7 +477,9 @@ class TrinerModelVITS:
         #config = self.training_args.to_dict()
         wandb.init(project= 'HugfaceTraining')
    
-    
+    def load_modell(self,namemodel):
+         self.model=VitsModel.from_pretrained(namemodel,token=self.token).to(self.device)
+         return "true"
     def load_dataset(self,train_dataset_dirs,eval_dataset_dir,full_generation_dir):
        ctrain_datasets,eval_dataset,full_generation_dataset=get_data_loader(train_dataset_dirs = train_dataset_dirs,
                                                               eval_dataset_dir =eval_dataset_dir ,
@@ -669,7 +671,8 @@ def init_Starting():
 def init_wandb():
     pro.init_wandb()
     return 'init_wandb'
-
+def loadd_d():
+    pro.load_modell("wasmdashai/vits-ar-sa-huba")
 def save_pretrained(path):
     pro.save_pretrained(path)