txya900619 commited on
Commit
2e5681a
·
1 Parent(s): e89a7e6

feat: use zero gpu

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,11 +1,13 @@
1
  import gradio as gr
2
  import numpy as np
 
3
 
4
  from ipa import g2p
5
  from ipa.ipa import text_to_ipa
6
  from models import models_config
7
 
8
 
 
9
  def _do_tts(model_id, ipa, language_name, speaker_name=None, speaker_wav=None):
10
  model = models_config[model_id]["model"]
11
  if speaker_wav is not None:
 
1
  import gradio as gr
2
  import numpy as np
3
+ import spaces
4
 
5
  from ipa import g2p
6
  from ipa.ipa import text_to_ipa
7
  from models import models_config
8
 
9
 
10
+ @spaces.GPU
11
  def _do_tts(model_id, ipa, language_name, speaker_name=None, speaker_wav=None):
12
  model = models_config[model_id]["model"]
13
  if speaker_wav is not None: