Update app.py

app.py

@@ -1,19 +1,19 @@
 # app.py
-import os, gradio as gr
+import os, gradio as gr, spaces
 from huggingface_hub import snapshot_download
 from indextts.infer import IndexTTS
 
 model_dir = snapshot_download("IndexTeam/IndexTTS-1.5", local_dir="checkpoints", local_dir_use_symlinks=False)
-cfg_path = os.path.join(model_dir, "config.yaml")
+cfg_path  = os.path.join(model_dir, "config.yaml")
 
 tts = None
-def _ensure_loaded():
+def load():
 	global tts
-	if tts is None:
-		tts = IndexTTS(model_dir=model_dir, cfg_path=cfg_path)
+	if tts is None: tts = IndexTTS(model_dir=model_dir, cfg_path=cfg_path)
 
+@spaces.GPU
 def synth(ref_wav, prompt):
-	_ensure_loaded()
+	load()
 	out = "out.wav"
 	tts.infer(ref_wav, prompt, out)
 	return out
@@ -22,8 +22,8 @@ with gr.Blocks() as demo:
 	gr.Markdown("# index-tts 1.5 zerogpu")
 	txt  = gr.Textbox(label="text prompt")
 	ref  = gr.Audio(label="reference voice", type="filepath")
-	out  = gr.Audio(label="generated speech", type="filepath")
-	gr.Button("generate").click(synth, [ref, txt], out)
+	gen  = gr.Audio(label="generated speech", type="filepath")
+	gr.Button("generate").click(synth, [ref, txt], gen)
 
 demo.queue()
-demo.launch(show_api=False, ssr_mode=False, max_file_size=50*1024*1024)
+demo.launch(show_api=False, ssr_mode=False)