Update app.py

app.py

@@ -2,27 +2,45 @@ import argparse
 import subprocess
 
 def main():
-    # 创建命令行参数解析器
-    parser = argparse.ArgumentParser(description="启动 Fish Speech 应用")
-    parser.add_argument("--llama-checkpoint-path", type=str, required=True, help="Llama 检查点路径")
-    parser.add_argument("--decoder-checkpoint-path", type=str, required=True, help="解码器检查点路径")
-    parser.add_argument("--decoder-config-name", type=str, required=True, help="解码器配置名称")
-    parser.add_argument("--device", type=str, default="cpu", help="设备类型（cpu 或 cuda）")
+    # 设置命令行参数解析器
+    parser = argparse.ArgumentParser(description="启动 WebUI")
+    parser.add_argument(
+        "--llama-checkpoint-path",
+        type=str,
+        default="checkpoints/fish-speech-1.4-sft-yth-lora",
+        help="Llama 检查点路径",
+    )
+    parser.add_argument(
+        "--decoder-checkpoint-path",
+        type=str,
+        default="checkpoints/fish-speech-1.4/firefly-gan-vq-fsq-8x1024-21hz-generator.pth",
+        help="解码器检查点路径",
+    )
+    parser.add_argument(
+        "--decoder-config-name",
+        type=str,
+        default="firefly_gan_vq",
+        help="解码器配置名称",
+    )
+    parser.add_argument(
+        "--device",
+        type=str,
+        default="cpu",
+        help="设备类型",
+    )
 
-    # 解析参数
+    # 解析命令行参数
     args = parser.parse_args()
 
-    # 构造命令
-    command = [
-        "python", "tools/webui.py",
+    # 启动 WebUI
+    subprocess.run([
+        "python",
+        "tools/webui.py",
         "--llama-checkpoint-path", args.llama_checkpoint_path,
         "--decoder-checkpoint-path", args.decoder_checkpoint_path,
         "--decoder-config-name", args.decoder_config_name,
-        "--device", args.device
-    ]
-
-    # 运行命令
-    subprocess.run(command)
+        "--device", args.device,
+    ])
 
 if __name__ == "__main__":
-    main()
+    main()