debug for hf space

indextts/infer.py

@@ -1,4 +1,3 @@
-import spaces
 import os
 import re
 import sys
@@ -16,7 +15,6 @@ from indextts.BigVGAN.models import BigVGAN as Generator
 
 
 class IndexTTS:
-    @spaces.GPU(duration=65)
     def __init__(self, cfg_path='checkpoints/config.yaml', model_dir='checkpoints'):
         self.cfg = OmegaConf.load(cfg_path)
         self.device = 'cuda:0'

webui.py

@@ -18,12 +18,13 @@ from tools.i18n.i18n import I18nAuto
 i18n = I18nAuto(language="zh_CN")
 MODE = 'local'
 snapshot_download("IndexTeam/Index-TTS",local_dir="checkpoints",)
-tts = IndexTTS(model_dir="checkpoints",cfg_path="checkpoints/config.yaml")
+tts = None
 
 os.makedirs("outputs/tasks",exist_ok=True)
 os.makedirs("prompts",exist_ok=True)
 
 def infer(voice, text,output_path=None):
+
     if not output_path:
         output_path = os.path.join("outputs", f"spk_{int(time.time())}.wav")
     tts.infer(voice, text, output_path)
@@ -67,7 +68,14 @@ with gr.Blocks() as demo:
                      inputs=[prompt_audio, input_text_single],
                      outputs=[output_audio])
 
-
-if __name__ == "__main__":
+@spaces.GPU
+def main():
+    global tts
+    if not tts:
+        tts = IndexTTS(model_dir="checkpoints", cfg_path="checkpoints/config.yaml")
     demo.queue(20)
     demo.launch(server_name="0.0.0.0")
+
+if __name__ == "__main__":
+    main()
+