Update generate_audio.py

generate_audio.py

@@ -19,7 +19,7 @@ class TTSGenerator:
     A class to generate podcast-style audio from a transcript using ParlerTTS and Bark models.
     """
     @spaces.GPU
-    def __init__(self, transcript_file_path):
+    def __init__(self, transcript_file_path,output_audio_path):
         """
         Initialize the TTS generator with the path to the rewritten transcript file.
         
@@ -27,7 +27,7 @@ class TTSGenerator:
             transcript_file_path (str): Path to the file containing the rewritten transcript.
         """
         self.transcript_file_path = transcript_file_path
-        self.output_audio_path = './resources/_podcast.mp3'
+        self.output_audio_path = output_audio_path
         
         # Set device
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -109,7 +109,8 @@ class TTSGenerator:
         wavfile.write(byte_io, sampling_rate, audio_int16)
         byte_io.seek(0)
         return AudioSegment.from_wav(byte_io)
-
+    
+    @spaces.GPU
     def generate_audio(self):
         """
         Converts the transcript into audio and saves it to a file.