debug for hf space

indextts/infer.py

@@ -1,3 +1,4 @@
+import spaces
 import os
 import re
 import sys
@@ -16,6 +17,7 @@ from indextts.BigVGAN.models import BigVGAN as Generator
 
 
 class IndexTTS:
+    @spaces.GPU
     def __init__(self, cfg_path='checkpoints/config.yaml', model_dir='checkpoints'):
         self.cfg = OmegaConf.load(cfg_path)
         self.device = 'cuda:0'
@@ -49,6 +51,7 @@ class IndexTTS:
     def preprocess_text(self, text):
         return self.normalizer.infer(text)
 
+    @spaces.GPU
     def infer(self, audio_prompt, text, output_path):
         text = self.preprocess_text(text)
 

webui.py

@@ -23,7 +23,7 @@ tts = None
 os.makedirs("outputs/tasks",exist_ok=True)
 os.makedirs("prompts",exist_ok=True)
 
-@spaces.GPU
+
 def infer(voice, text,output_path=None):
     global tts
     if not tts: