TGPro1 commited on
Commit
831d6bc
·
verified ·
1 Parent(s): 569c1ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -11
app.py CHANGED
@@ -6,19 +6,25 @@ import torch
6
  os.environ["COQUI_TOS_AGREED"] = "1"
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
 
9
- tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
 
 
 
 
 
 
 
10
 
11
  def clone(text, lang, ref):
12
- out = "out.wav"
13
  tts.tts_to_file(text=text, file_path=out, speaker_wav=ref, language=lang)
14
  return out
15
 
16
- LANGS = ["en","es","fr","de","it","pt","pl","tr","ru","nl","cs","ar","zh-cn","ja","ko","hu"]
17
-
18
- with gr.Blocks() as demo:
19
- t=gr.Textbox(label="Text")
20
- l=gr.Dropdown(label="Lang", choices=LANGS, value="fr")
21
- r=gr.Audio(label="Ref", type="filepath")
22
- o=gr.Audio(label="Out")
23
- gr.Button("Generate").click(clone, [t,l,r], [o])
24
- demo.launch()
 
6
  os.environ["COQUI_TOS_AGREED"] = "1"
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
 
9
+ print("Loading XTTS...")
10
+ try:
11
+ # Try direct ID loading which works on most versions
12
+ tts = TTS(model_name="tts_models/multilingual/multi-dataset/xtts_v2", progress_bar=False).to(device)
13
+ except Exception as e:
14
+ print(f"Error loading by name: {e}")
15
+ # Fallback to general initialization which triggers download
16
+ tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
17
 
18
  def clone(text, lang, ref):
19
+ out = "out.wav"
20
  tts.tts_to_file(text=text, file_path=out, speaker_wav=ref, language=lang)
21
  return out
22
 
23
+ LANGS = ["en","es","fr","de","it","pt","pl","tr","ru","nl","cs","ar","zh-cn","ja","ko","hu"
24
+ with gr.Blocks() as demo:
25
+ t=gr.Textbox(label="Text")
26
+ l=gr.Dropdown(label="Lang", choices=LANGS, value="fr")
27
+ r=gr.Audio(label="Ref", type="filepath")
28
+ o=gr.Audio(label="Out")
29
+ gr.Button("Generate").click(clone, [t,l,r], [o])
30
+ demo.launch()]