Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,8 @@ import tempfile
|
|
| 10 |
import torchaudio
|
| 11 |
|
| 12 |
os.environ["COQUI_TOS_AGREED"] = "1"
|
| 13 |
-
device = "cuda"
|
|
|
|
| 14 |
|
| 15 |
tts = TTS("tts_models/multilingual/multi-dataset/xtts_bill_spa").to(device)
|
| 16 |
model_path = '/home/user/.local/share/tts/tts_models--multilingual--multi-dataset--xtts_bill_spa/model.pth'
|
|
@@ -26,9 +27,8 @@ def load_model(xtts_checkpoint, xtts_config, xtts_vocab):
|
|
| 26 |
XTTS_MODEL = Xtts.init_from_config(config)
|
| 27 |
print("Loading XTTS model! ")
|
| 28 |
XTTS_MODEL.load_checkpoint(config, checkpoint_path=xtts_checkpoint, vocab_path=xtts_vocab, use_deepspeed=False)
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
print("Model Loaded!")
|
| 33 |
|
| 34 |
def run_tts(lang, tts_text, speaker_audio_file):
|
|
|
|
| 10 |
import torchaudio
|
| 11 |
|
| 12 |
os.environ["COQUI_TOS_AGREED"] = "1"
|
| 13 |
+
#device = "cuda" # Comentar esta línea para usar CPU en lugar de GPU
|
| 14 |
+
device = "cpu" # Especificar CPU como dispositivo
|
| 15 |
|
| 16 |
tts = TTS("tts_models/multilingual/multi-dataset/xtts_bill_spa").to(device)
|
| 17 |
model_path = '/home/user/.local/share/tts/tts_models--multilingual--multi-dataset--xtts_bill_spa/model.pth'
|
|
|
|
| 27 |
XTTS_MODEL = Xtts.init_from_config(config)
|
| 28 |
print("Loading XTTS model! ")
|
| 29 |
XTTS_MODEL.load_checkpoint(config, checkpoint_path=xtts_checkpoint, vocab_path=xtts_vocab, use_deepspeed=False)
|
| 30 |
+
# Mover el modelo a la CPU si se está usando CPU
|
| 31 |
+
XTTS_MODEL.to(device) # Esto asegura que el modelo se cargue en la CPU
|
|
|
|
| 32 |
print("Model Loaded!")
|
| 33 |
|
| 34 |
def run_tts(lang, tts_text, speaker_audio_file):
|