debug for hf space

indextts/infer.py

@@ -44,6 +44,9 @@ class IndexTTS:
         self.bigvgan = self.bigvgan.to(self.device)
         self.bigvgan.eval()
         print(">> bigvgan weights restored from:", self.bigvgan_path)
+        self.normalizer = None
+
+    def load_normalizer(self):
         self.normalizer = TextNormalizer()
         self.normalizer.load()
         print(">> TextNormalizer loaded")
@@ -156,4 +159,5 @@ class IndexTTS:
 
 if __name__ == "__main__":
     tts = IndexTTS(cfg_path="checkpoints/config.yaml", model_dir="checkpoints")
+    tts.load_normalizer()
     tts.infer(audio_prompt='test_data/input.wav', text='大家好，我现在正在bilibili 体验 ai 科技，说实话，来之前我绝对想不到！AI技术已经发展到这样匪夷所思的地步了！',output_path="gen.wav")

webui.py

@@ -28,6 +28,7 @@ def infer(voice, text,output_path=None):
     global tts
     if not tts:
         tts = IndexTTS(model_dir="checkpoints", cfg_path="checkpoints/config.yaml")
+        tts.load_normalizer()
     if not output_path:
         output_path = os.path.join("outputs", f"spk_{int(time.time())}.wav")
     tts.infer(voice, text, output_path)
@@ -76,6 +77,7 @@ def main():
     global tts
     if not tts:
         tts = IndexTTS(model_dir="checkpoints", cfg_path="checkpoints/config.yaml")
+        tts.load_normalizer()
     demo.queue(20)
     demo.launch(server_name="0.0.0.0")