Chin-Yun Yu commited on
Commit
211cca8
·
1 Parent(s): a2cb94b

fix: init cuda in main

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -513,7 +513,7 @@ def vec2fx(x):
513
  return fx
514
 
515
 
516
- @spaces.GPU(duration=10)
517
  def infer_chain(input_audio, remove_approx, ratio, *args):
518
  return chain_functions(
519
  lambda audio, approx, ratio, *args: (
@@ -729,8 +729,10 @@ with gr.Blocks() as demo:
729
  )
730
 
731
  remove_approx_checkbox.change(
732
- lambda in_audio, out_audio, di, wet, *args: (
733
- (out_audio, di, wet) if in_audio is None else render(in_audio, *args)
 
 
734
  ),
735
  inputs=[
736
  audio_input,
 
513
  return fx
514
 
515
 
516
+ @spaces.GPU(duration=60)
517
  def infer_chain(input_audio, remove_approx, ratio, *args):
518
  return chain_functions(
519
  lambda audio, approx, ratio, *args: (
 
729
  )
730
 
731
  remove_approx_checkbox.change(
732
+ spaces.GPU(duration=10)(
733
+ lambda in_audio, out_audio, di, wet, *args: (
734
+ (out_audio, di, wet) if in_audio is None else render(in_audio, *args)
735
+ )
736
  ),
737
  inputs=[
738
  audio_input,