Approximetal commited on
Commit
c7f69cf
·
verified ·
1 Parent(s): f45ef87

Update gradio_mix.py

Browse files
Files changed (1) hide show
  1. gradio_mix.py +4 -4
gradio_mix.py CHANGED
@@ -700,7 +700,7 @@ def get_transcribe_state(segments):
700
  "word_bounds": [f"{word['start']} {word['word']} {word['end']}" for word in segments["words"]]
701
  }
702
 
703
- @spaces.GPU
704
  @torch.no_grad()
705
  @torch.inference_mode()
706
  def transcribe(seed, audio_info):
@@ -719,7 +719,7 @@ def transcribe(seed, audio_info):
719
  state
720
  ]
721
 
722
- @spaces.GPU
723
  @torch.no_grad()
724
  @torch.inference_mode()
725
  def align(transcript, audio_info, state):
@@ -747,7 +747,7 @@ def align(transcript, audio_info, state):
747
  ]
748
 
749
 
750
- @spaces.GPU
751
  @torch.no_grad()
752
  @torch.inference_mode()
753
  def denoise(audio_info):
@@ -796,7 +796,7 @@ def replace_numbers_with_words(sentence, lang="en"):
796
  return re.sub(r'\b\d+\b', replace_with_words, sentence) # Regular expression that matches numbers
797
 
798
 
799
- @spaces.GPU
800
  @torch.no_grad()
801
  @torch.inference_mode()
802
  def run(seed, nfe_step, speed, cfg_strength, sway_sampling_coef, ref_ratio,
 
700
  "word_bounds": [f"{word['start']} {word['word']} {word['end']}" for word in segments["words"]]
701
  }
702
 
703
+ @spaces.GPU(duration=240)
704
  @torch.no_grad()
705
  @torch.inference_mode()
706
  def transcribe(seed, audio_info):
 
719
  state
720
  ]
721
 
722
+ @spaces.GPU(duration=240)
723
  @torch.no_grad()
724
  @torch.inference_mode()
725
  def align(transcript, audio_info, state):
 
747
  ]
748
 
749
 
750
+ @spaces.GPU(duration=240)
751
  @torch.no_grad()
752
  @torch.inference_mode()
753
  def denoise(audio_info):
 
796
  return re.sub(r'\b\d+\b', replace_with_words, sentence) # Regular expression that matches numbers
797
 
798
 
799
+ @spaces.GPU(duration=240)
800
  @torch.no_grad()
801
  @torch.inference_mode()
802
  def run(seed, nfe_step, speed, cfg_strength, sway_sampling_coef, ref_ratio,