|
import torch,glob |
|
from maha_tts import load_diffuser,load_models,infer_tts,config |
|
from scipy.io.wavfile import write |
|
|
|
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') |
|
print('Using:',device) |
|
text = 'Printing, in the only sense with which we are at present concerned, differs from most if not from all the arts and crafts represented in the Exhibition.' |
|
langauge = 'english' |
|
language = torch.tensor(config.lang_index[langauge]).to(device).unsqueeze(0) |
|
ref_clips = glob.glob('models/Smolie-en/ref_clips/part0_1_1/*.wav') |
|
|
|
|
|
|
|
diff_model,ts_model,vocoder,diffuser = load_models('Smolie-in',device) |
|
audio,sr = infer_tts(text,ref_clips,diffuser,diff_model,ts_model,vocoder,language) |
|
write('test.wav',sr,audio) |
