Rename webui_colab.py to app.py

webui_colab.py → app.py

@@ -30,6 +30,8 @@ logging.getLogger('matplotlib').setLevel(logging.WARNING)
 from cosyvoice.cli.cosyvoice import CosyVoice
 from cosyvoice.utils.file_utils import load_wav
 
+import spaces
+
 logging.basicConfig(level=logging.WARNING,
                     format='%(asctime)s %(levelname)s %(message)s')
 
@@ -66,6 +68,7 @@ instruct_dict = {'预训练音色': '1. 选择预训练音色\n2.点击生成音
 def change_instruction(mode_checkbox_group):
     return instruct_dict[mode_checkbox_group]
 
+@spaces.GPU
 def generate_audio(tts_text, mode_checkbox_group, sft_dropdown, prompt_text, prompt_wav_upload, prompt_wav_record, instruct_text, seed):
     tts_text = "".join([item for item in tts_text.strip().split("\n") if item != ""]) + ".。"
     if prompt_wav_upload is not None:
@@ -173,8 +176,8 @@ def main():
                         </p>
             </div>
         ''')
-    demo.queue()
-    demo.launch(share=True, show_error=True)
+    demo.queue(max_size=40, api_open=False)
+    demo.launch(max_threads=400, show_error=True)
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()