simler commited on
Commit
c733108
·
verified ·
1 Parent(s): af13e4e

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -0
app.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import sys
3
+ import logging
4
+
5
+ # 配置日志,方便出问题时看 Space 的 Logs
6
+ logging.basicConfig(level=logging.INFO)
7
+ logger = logging.getLogger(__name__)
8
+
9
+ # 尝试导入核心组件
10
+ try:
11
+ from genie_tts.Server import Server
12
+ # 根据新版结构,TTS 推理引擎通常在 Core.Inference
13
+ from genie_tts.Core.Inference import TTS
14
+ logger.info("✅ 成功导入 Genie-TTS 模块")
15
+ except ImportError as e:
16
+ logger.error(f"❌ 导入失败: {e}")
17
+ sys.exit(1)
18
+
19
+ def main():
20
+ # 1. 初始化推理引擎
21
+ # model_dir="." 表示读取当前目录下的 onnx 文件
22
+ logger.info("正在加载 TTS 模型,路径: .")
23
+ try:
24
+ engine = TTS(model_dir=".")
25
+ except Exception as e:
26
+ logger.error(f"❌ 模型加载失败: {e}")
27
+ # 有时候是因为缺少缓存目录,手动创建一下
28
+ os.makedirs("cache", exist_ok=True)
29
+ engine = TTS(model_dir=".")
30
+
31
+ # 2. 启动 API 服务
32
+ # Hugging Face Space 强制要求监听 0.0.0.0 和端口 7860
33
+ logger.info("启动 API 服务 (Port 7860)...")
34
+ server = Server(instance=engine, host="0.0.0.0", port=7860)
35
+ server.run()
36
+
37
+ if __name__ == "__main__":
38
+ main()