Hugging Face's logo

Spaces:
Bai-YT
/
ConsistencyTTA
Running on Zero

App Files Community
Bai-YT commited on
Commit
8310070
•
1 Parent(s): 3e3c49b

Update run_gradio.py

Browse files
Files changed (1) hide show
  1. run_gradio.py +1 -0
run_gradio.py CHANGED
@@ -33,6 +33,7 @@ consistencytta.eval()
33
  consistencytta.requires_grad_(False)
34
 
35
 
 
36
  def generate(prompt: str, seed: str = '', cfg_weight: float = 4.):
37
  """ Generate audio from a given prompt.
38
  Args:
 
33
  consistencytta.requires_grad_(False)
34
 
35
 
36
+ @spaces.GPU()
37
  def generate(prompt: str, seed: str = '', cfg_weight: float = 4.):
38
  """ Generate audio from a given prompt.
39
  Args: