Create app.py

app.py

@@ -18,8 +18,11 @@ st.success("Models loaded successfully!")
 def text_to_speech(text):
     # Generate mel spectrogram
     mel_output, mel_length, alignment = tacotron2.encode_text(text)
+    
     # Decode mel spectrogram to waveform
+    # FIX: hifi_gan.decode_batch() returns only 1 value
     waveforms = hifi_gan.decode_batch(mel_output)
+    
     # Save waveform as audio file
     audio_path = "output.wav"
     write(audio_path, 22050, waveforms.squeeze(1).cpu().numpy())