r3gm commited on
Commit
d86cb25
1 Parent(s): e7d1ed8

Update soni_translate/speech_segmentation.py

Browse files
soni_translate/speech_segmentation.py CHANGED
@@ -27,7 +27,7 @@ def random_sleep():
27
  time.sleep(sleep_time)
28
 
29
 
30
- @spaces.GPU(duration=120)
31
  def load_and_transcribe_audio(asr_model, audio, compute_type, language, asr_options, batch_size, segment_duration_limit):
32
  # Load model
33
  model = whisperx.load_model(
@@ -81,7 +81,7 @@ def load_align_and_align_segments(result, audio, DAMHF):
81
 
82
  return alignment_result
83
 
84
- @spaces.GPU(duration=120)
85
  def diarize_audio(diarize_model, audio_wav, min_speakers, max_speakers):
86
 
87
  if os.environ.get("ZERO_GPU") == "TRUE":
 
27
  time.sleep(sleep_time)
28
 
29
 
30
+ @spaces.GPU(duration=110)
31
  def load_and_transcribe_audio(asr_model, audio, compute_type, language, asr_options, batch_size, segment_duration_limit):
32
  # Load model
33
  model = whisperx.load_model(
 
81
 
82
  return alignment_result
83
 
84
+ @spaces.GPU(duration=110)
85
  def diarize_audio(diarize_model, audio_wav, min_speakers, max_speakers):
86
 
87
  if os.environ.get("ZERO_GPU") == "TRUE":