Naozumi0512 commited on
Commit
deea864
1 Parent(s): e62fb95
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  # flake8: noqa: E402
 
2
  import os
3
  import logging
4
  import re_matching
@@ -32,6 +33,7 @@ if device == "mps":
32
  os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
33
 
34
 
 
35
  def generate_audio(
36
  slices,
37
  sdp_ratio,
 
1
  # flake8: noqa: E402
2
+ import spaces
3
  import os
4
  import logging
5
  import re_matching
 
33
  os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
34
 
35
 
36
+ @spaces.GPU
37
  def generate_audio(
38
  slices,
39
  sdp_ratio,