Update tts.py

tts.py

@@ -25,12 +25,13 @@ def synthesize_speech(text):
         result = pipe(text)
         logging.debug(f"Pipeline result: {result}")
 
-        # Check if the output contains 'waveform' key
-        if 'waveform' in result:
-            waveform = result['waveform']
-            # Save waveform to an audio file
+        # Check if the output contains 'audio' and 'sampling_rate' keys
+        if 'audio' in result and 'sampling_rate' in result:
+            audio = result['audio']
+            sampling_rate = result['sampling_rate']
+            # Save audio to an audio file
             audio_path = "output.wav"
-            sf.write(audio_path, waveform, 16000)  # Write the waveform using soundfile
+            sf.write(audio_path, audio[0], sampling_rate)  # Write the audio using soundfile
             return audio_path
         else:
             logging.error(f"Unexpected pipeline output: {result}")