Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|