Update deploy.py

deploy.py

@@ -1,10 +1,6 @@
 import logging
-logging.basicConfig(level=logging.INFO)
-
-import spaces
-import os
-import sys
 import subprocess
+logging.basicConfig(level=logging.INFO)
 
 def run_shell_script(script_path):
     """
@@ -29,59 +25,17 @@ def run_shell_script(script_path):
 # 假设有一个名为example.sh的脚本文件在当前目录下
 run_shell_script('deploy.sh')
 
-os.symlink("/usr/lib/x86_64-linux-gnu/libsox.so.3", "/usr/lib/x86_64-linux-gnu/libsox.so")
-
-ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
-sys.path.append('{}/third_party/AcademiCodec'.format(ROOT_DIR))
-sys.path.append('{}/third_party/Matcha-TTS'.format(ROOT_DIR))
-
-os.system('cd pretrained_models/speech_kantts_ttsfrd/ && pip install ttsfrd_dependency-0.1-py3-none-any.whl')
-os.system('cd pretrained_models/speech_kantts_ttsfrd/ && pip install ttsfrd-0.3.9-cp310-cp310-linux_x86_64.whl')
-os.system('sed -i [email protected]@typing_extensions@g /usr/local/lib/python3.10/site-packages/inflect/__init__.py')
-
-import gradio as gr
-from css.advanced import advanced
-from css.custom import custom
-from css.preset import preset
-
-audio_mode_choices = [('Preset voice generation', 'preset'), ('Custom voice generation (replicate recorded voice)', 'custom'),
-                      ('Advanced voice generation (natural language control)', 'advanced')]
-
-
-
-def on_audio_mode_change(_audio_mode_radio):
-    yield {
-        preset_layout: gr.update(visible=_audio_mode_radio == 'preset'),
-        custom_layout: gr.update(visible=_audio_mode_radio == 'custom'),
-        advanced_layout: gr.update(visible=_audio_mode_radio == 'advanced')
-    }
-
-
-custom_css = """
-.full-height {
-    height: 100%;
-}
-"""
-
-default_layout = 'preset'
-
-with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
-    audio_mode_radio = gr.Radio(choices=audio_mode_choices,
-                                value=default_layout,
-                                label="Select voice generation mode")
-    with gr.Row():
-        with gr.Column(visible=default_layout == 'preset') as preset_layout:
-            preset()
-        with gr.Column(visible=default_layout == 'custom') as custom_layout:
-            custom()
-        with gr.Column(
-                visible=default_layout == 'advanced') as advanced_layout:
-            advanced()
-
-    audio_mode_radio.change(
-        fn=on_audio_mode_change,
-        inputs=[audio_mode_radio],
-        outputs=[preset_layout, custom_layout, advanced_layout])
+class args:
+    def __init__(self):
+        self.port = 5000
 
+from CosyVoice.webui import main
+from CosyVoice.cosyvoice.cli.cosyvoice import CosyVoice
+import numpy as np
 
-demo.queue().launch(server_port=7860)
+cosyvoice = CosyVoice("pretrained_models/CosyVoice-300M")
+sft_spk = cosyvoice.list_avaliable_spks()
+prompt_sr, target_sr = 16000, 22050
+default_data = np.zeros(target_sr)
+args.port = 5000
+main()