Spaces:

tester1hf
/

tests

Sleeping

tester1hf commited on Feb 18, 2025

Commit

dc4db49

verified ·

1 Parent(s): a60e434

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ torch.load = patched_torch_load
 # Initialize XTTS model
 device = "cuda" if torch.cuda.is_available() else "cpu"
-tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=True)
 def extract_speaker_embedding(audio_path):
     try:

 # Initialize XTTS model
 device = "cuda" if torch.cuda.is_available() else "cpu"
+tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
 def extract_speaker_embedding(audio_path):
     try: