wasmdashai commited on
Commit
609de06
·
verified ·
1 Parent(s): 452956c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +32 -0
app.py CHANGED
@@ -540,6 +540,38 @@ def modelspeech(texts):
540
  wav = model_vits(input_ids=inputs["input_ids"]).waveform#.detach()
541
  # display(Audio(wav, rate=model.config.sampling_rate))
542
  return model_vits.config.sampling_rate,wav#remove_noise_nr(wav)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
543
  def greet(text,id):
544
  global GK
545
  b=int(id)
 
540
  wav = model_vits(input_ids=inputs["input_ids"]).waveform#.detach()
541
  # display(Audio(wav, rate=model.config.sampling_rate))
542
  return model_vits.config.sampling_rate,wav#remove_noise_nr(wav)
543
+
544
+ dataset_dir='ABThag-db'
545
+ train_dataset_dirs=[
546
+ # ('/content/drive/MyDrive/vitsM/DATA/fahd_db',0),
547
+ # ('/content/drive/MyDrive/vitsM/DATA/fahd_db',0),
548
+ # ('/content/drive/MyDrive/vitsM/DB2KKKK',1),
549
+ # ('/content/drive/MyDrive/vitsM/DATA/Db_Amgd_50_Bitch10',0),
550
+ # ('/content/drive/MyDrive/vitsM/DB2KKKK',1), #
551
+ # ('/content/drive/MyDrive/vitsM/DATA/Db_Amgd_50_Bitch10',0),
552
+ # ('/content/drive/MyDrive/vitsM/DATA/DBWfaa-Bitch:8-Count:60',0),
553
+ # ('/content/drive/MyDrive/vitsM/DATA/Wafa/b10r',0),
554
+ # ('/content/drive/MyDrive/vitsM/DATA/Wafa/b16r',0),
555
+ # ('/content/drive/MyDrive/vitsM/DATA/Wafa/b4',0),
556
+
557
+ # ('/content/drive/MyDrive/vitsM/DATA/fahd_db',None),
558
+ # ('/content/drive/MyDrive/vitsM/DATA/wafa-db',None),
559
+ # ('/content/drive/MyDrive/vitsM/DATA/wafa-db',4),
560
+ # ('/content/drive/MyDrive/vitsM/DATA/DB-ABThag-Bitch:5-Count-37',4),
561
+ # ('/content/drive/MyDrive/vitsM/DB-300-k',6),
562
+ ('ABThag-db',0),
563
+ #('/content/drive/MyDrive/dataset_ljBatchs',0),
564
+
565
+
566
+
567
+
568
+
569
+ ]
570
+
571
+ ctrain_datasets,eval_dataset,full_generation_dataset=get_data_loader(train_dataset_dirs = train_dataset_dirs,
572
+ eval_dataset_dir = os.path.join(dataset_dir,'eval'),
573
+ full_generation_dir = os.path.join(dataset_dir,'full_generation'),
574
+ device=device)
575
  def greet(text,id):
576
  global GK
577
  b=int(id)