chong.zhang commited on
Commit
c3b6520
·
1 Parent(s): 9c6a7bd
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,7 @@
13
  # limitations under the License.
14
 
15
  import os
16
- # import spaces
17
  import gradio as gr
18
  from inspiremusic.cli.inference import InspireMusicUnified, set_env_variables
19
  import torchaudio
@@ -24,6 +24,7 @@ import subprocess
24
  import importlib
25
  import sys
26
 
 
27
  os.system('nvidia-smi')
28
  os.system('apt update -y && apt-get install -y apt-utils && apt install -y unzip')
29
  os.system('pip install flash-attn --no-build-isolation')
 
13
  # limitations under the License.
14
 
15
  import os
16
+ import spaces
17
  import gradio as gr
18
  from inspiremusic.cli.inference import InspireMusicUnified, set_env_variables
19
  import torchaudio
 
24
  import importlib
25
  import sys
26
 
27
+ @spaces.GPU
28
  os.system('nvidia-smi')
29
  os.system('apt update -y && apt-get install -y apt-utils && apt install -y unzip')
30
  os.system('pip install flash-attn --no-build-isolation')