Bmo411 commited on
Commit
289e144
·
verified ·
1 Parent(s): 7cbab11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -27,11 +27,10 @@ generator = keras.models.load_model(model_path, compile=False)
27
  def text_to_audio(text):
28
  # 1. Convertir texto a mel-spectrograma
29
  mel_output, _, _ = tacotron2.encode_text(text)
30
- mel = mel_output.squeeze(0).detach().cpu().numpy().astype(np.float32) # (80, frames)
31
-
32
  # 2. Preparar para generator
33
- mel_input = mel[np.newaxis, ..., np.newaxis] # (1, 80, frames, 1)
34
- mel_input = tf.convert_to_tensor(mel_input)
35
 
36
  # 3. Usar generator para generar audio
37
  fake_audio = generator(mel_input, training=False)
 
27
  def text_to_audio(text):
28
  # 1. Convertir texto a mel-spectrograma
29
  mel_output, _, _ = tacotron2.encode_text(text)
30
+ mel = mel_output.detach().cpu().numpy().astype(np.float32) # (80, frames)
31
+
32
  # 2. Preparar para generator
33
+ mel_input = np.expand_dims(mel,axis=-1) # (1, 80, frames, 1)
 
34
 
35
  # 3. Usar generator para generar audio
36
  fake_audio = generator(mel_input, training=False)