yasserrmd commited on
Commit
950f7ad
·
verified ·
1 Parent(s): 083ec97

Update generate_audio.py

Browse files
Files changed (1) hide show
  1. 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.