Spaces:
Running
Running
Update generate_audio.py
Browse files- generate_audio.py +2 -2
generate_audio.py
CHANGED
@@ -59,7 +59,7 @@ class TTSGenerator:
|
|
59 |
with open(self.transcript_file_path, 'rb') as f:
|
60 |
return ast.literal_eval(pickle.load(f))
|
61 |
|
62 |
-
@spaces.GPU
|
63 |
def generate_speaker1_audio(self, text):
|
64 |
"""
|
65 |
Generate audio for Speaker 1 using ParlerTTS.
|
@@ -92,7 +92,7 @@ class TTSGenerator:
|
|
92 |
audio_arr = generation.cpu().numpy().squeeze()
|
93 |
return audio_arr, self.parler_model.config.sampling_rate
|
94 |
|
95 |
-
@spaces.GPU
|
96 |
def generate_speaker2_audio(self, text):
|
97 |
"""
|
98 |
Generate audio for Speaker 2 using Bark.
|
|
|
59 |
with open(self.transcript_file_path, 'rb') as f:
|
60 |
return ast.literal_eval(pickle.load(f))
|
61 |
|
62 |
+
@spaces.GPU(duration=30)
|
63 |
def generate_speaker1_audio(self, text):
|
64 |
"""
|
65 |
Generate audio for Speaker 1 using ParlerTTS.
|
|
|
92 |
audio_arr = generation.cpu().numpy().squeeze()
|
93 |
return audio_arr, self.parler_model.config.sampling_rate
|
94 |
|
95 |
+
@spaces.GPU(duration=30)
|
96 |
def generate_speaker2_audio(self, text):
|
97 |
"""
|
98 |
Generate audio for Speaker 2 using Bark.
|