Uhhy commited on
Commit
d2a98af
·
verified ·
1 Parent(s): ae36eb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- if torch.cuda.is_available():
30
- XTTS_MODEL.cuda()
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):