Spaces:

owiedotch
/

demucs-stem-separation

Running on Zero

owiedotch commited on Sep 1

Commit

0ff04fc

•

1 Parent(s): ac0c0ce

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,11 @@
 import gradio as gr
 import os
 import subprocess
 from typing import Tuple, List, Dict
 from pydub import AudioSegment
 def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass: bool, other: bool, mp3: bool, mp3_bitrate: int) -> Tuple[str, gr.HTML]:
     log_messages = []

 import gradio as gr
 import os
 import subprocess
+import spaces
 from typing import Tuple, List, Dict
 from pydub import AudioSegment
+@spaces.GPU
 def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass: bool, other: bool, mp3: bool, mp3_bitrate: int) -> Tuple[str, gr.HTML]:
     log_messages = []