junxiliu commited on
Commit
c847c87
·
1 Parent(s): 7fe1c8e
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
 
2
  import warnings
3
- #import spaces
4
  warnings.filterwarnings("ignore", category=FutureWarning)
5
  import logging
6
  from argparse import ArgumentParser
@@ -119,7 +119,7 @@ def load_model_if_needed(
119
  log.info(f"Model '{variant}' already loaded with current settings.")
120
  return False
121
 
122
- #@spaces.GPU
123
  @torch.inference_mode()
124
  def generate_audio_gradio(
125
  prompt,
 
1
 
2
  import warnings
3
+ import spaces
4
  warnings.filterwarnings("ignore", category=FutureWarning)
5
  import logging
6
  from argparse import ArgumentParser
 
119
  log.info(f"Model '{variant}' already loaded with current settings.")
120
  return False
121
 
122
+ @spaces.GPU
123
  @torch.inference_mode()
124
  def generate_audio_gradio(
125
  prompt,