Update app.py

app.py

@@ -540,6 +540,38 @@ def   modelspeech(texts):
      wav = model_vits(input_ids=inputs["input_ids"]).waveform#.detach()
           # display(Audio(wav, rate=model.config.sampling_rate))
      return  model_vits.config.sampling_rate,wav#remove_noise_nr(wav)
+
+dataset_dir='ABThag-db'
+train_dataset_dirs=[
+       #   ('/content/drive/MyDrive/vitsM/DATA/fahd_db',0),
+          # ('/content/drive/MyDrive/vitsM/DATA/fahd_db',0),
+        #   ('/content/drive/MyDrive/vitsM/DB2KKKK',1),
+        #  ('/content/drive/MyDrive/vitsM/DATA/Db_Amgd_50_Bitch10',0),
+            # ('/content/drive/MyDrive/vitsM/DB2KKKK',1), #
+        #  ('/content/drive/MyDrive/vitsM/DATA/Db_Amgd_50_Bitch10',0),
+          # ('/content/drive/MyDrive/vitsM/DATA/DBWfaa-Bitch:8-Count:60',0),
+      #  ('/content/drive/MyDrive/vitsM/DATA/Wafa/b10r',0),
+      #  ('/content/drive/MyDrive/vitsM/DATA/Wafa/b16r',0),
+      #  ('/content/drive/MyDrive/vitsM/DATA/Wafa/b4',0),
+
+          # ('/content/drive/MyDrive/vitsM/DATA/fahd_db',None),
+          # ('/content/drive/MyDrive/vitsM/DATA/wafa-db',None),
+          # ('/content/drive/MyDrive/vitsM/DATA/wafa-db',4),
+          # ('/content/drive/MyDrive/vitsM/DATA/DB-ABThag-Bitch:5-Count-37',4),
+         # ('/content/drive/MyDrive/vitsM/DB-300-k',6),
+         ('ABThag-db',0),
+         #('/content/drive/MyDrive/dataset_ljBatchs',0),
+
+
+
+
+
+                    ]
+
+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)
 def greet(text,id):
     global GK 
     b=int(id)