debug for hf space

webui.py

@@ -23,8 +23,11 @@ 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:
+        tts = IndexTTS(model_dir="checkpoints", cfg_path="checkpoints/config.yaml")
     if not output_path:
         output_path = os.path.join("outputs", f"spk_{int(time.time())}.wav")
     tts.infer(voice, text, output_path)
@@ -68,11 +71,8 @@ with gr.Blocks() as demo:
                      inputs=[prompt_audio, input_text_single],
                      outputs=[output_audio])
 
-@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")