whooray commited on
Commit
0a07e49
·
verified ·
1 Parent(s): 2039774

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,12 +1,13 @@
1
  import os
2
  import gradio as gr
3
  from scipy.io.wavfile import write
 
4
 
5
-
6
  def inference(audio):
7
  os.makedirs("out", exist_ok=True)
8
  write('test.wav', audio[0], audio[1])
9
- os.system("python3 -m demucs.separate -n mdx_extra_q --two-stems=vocals -d cpu test.wav -o out")
10
  return "./out/mdx_extra_q/test/vocals.wav","./out/mdx_extra_q/test/no_vocals.wav"
11
 
12
  title = "음성 분리"
 
1
  import os
2
  import gradio as gr
3
  from scipy.io.wavfile import write
4
+ import spaces
5
 
6
+ @spaces.GPU(duration=200)
7
  def inference(audio):
8
  os.makedirs("out", exist_ok=True)
9
  write('test.wav', audio[0], audio[1])
10
+ os.system("python3 -m demucs.separate -n mdx_extra_q --two-stems=vocals test.wav -o out")
11
  return "./out/mdx_extra_q/test/vocals.wav","./out/mdx_extra_q/test/no_vocals.wav"
12
 
13
  title = "음성 분리"