Merge pull request #210 from jiangyuxiaoxiao/master

update server_fastapi.py: 缺少web文件时添加提示信息。configyml添加更多帮助注释
This commit is contained in:
Sora
2023-12-01 10:14:24 +08:00
committed by GitHub
2 changed files with 19 additions and 9 deletions

View File

@@ -121,8 +121,10 @@ server:
port: 5000 port: 5000
# 模型默认使用设备:但是当前并没有实现这个配置。 # 模型默认使用设备:但是当前并没有实现这个配置。
device: "cuda" device: "cuda"
# 需要加载的所有模型的配置 # 需要加载的所有模型的配置,可以填多个模型,也可以不填模型,等网页成功后手动加载模型
# 不加载模型的配置格式删除默认给的两个模型配置给models赋值 [ ]也就是空列表。参考模型2的speakers 即 models: [ ]
# 注意所有模型都必须正确配置model与config的路径空路径会导致加载错误。 # 注意所有模型都必须正确配置model与config的路径空路径会导致加载错误。
# 也可以不填模型等网页加载成功后手动填写models。
models: models:
- # 模型的路径 - # 模型的路径
model: "" model: ""

View File

@@ -149,7 +149,13 @@ if __name__ == "__main__":
app = FastAPI() app = FastAPI()
app.logger = logger app.logger = logger
# 挂载静态文件 # 挂载静态文件
logger.info("开始挂载网页页面")
StaticDir: str = "./Web" StaticDir: str = "./Web"
if not os.path.isdir(StaticDir):
logger.warning(
"缺少网页资源,无法开启网页页面,如有需要请在 https://github.com/jiangyuxiaoxiao/Bert-VITS2-UI 或者Bert-VITS对应版本的release页面下载"
)
else:
dirs = [fir.name for fir in os.scandir(StaticDir) if fir.is_dir()] dirs = [fir.name for fir in os.scandir(StaticDir) if fir.is_dir()]
files = [fir.name for fir in os.scandir(StaticDir) if fir.is_dir()] files = [fir.name for fir in os.scandir(StaticDir) if fir.is_dir()]
for dirName in dirs: for dirName in dirs:
@@ -160,6 +166,7 @@ if __name__ == "__main__":
) )
loaded_models = Models() loaded_models = Models()
# 加载模型 # 加载模型
logger.info("开始加载模型")
models_info = config.server_config.models models_info = config.server_config.models
for model_info in models_info: for model_info in models_info:
loaded_models.init_model( loaded_models.init_model(
@@ -605,6 +612,7 @@ if __name__ == "__main__":
logger.warning("本地服务,请勿将服务端口暴露于外网") logger.warning("本地服务,请勿将服务端口暴露于外网")
logger.info(f"api文档地址 http://127.0.0.1:{config.server_config.port}/docs") logger.info(f"api文档地址 http://127.0.0.1:{config.server_config.port}/docs")
if os.path.isdir(StaticDir):
webbrowser.open(f"http://127.0.0.1:{config.server_config.port}") webbrowser.open(f"http://127.0.0.1:{config.server_config.port}")
uvicorn.run( uvicorn.run(
app, port=config.server_config.port, host="0.0.0.0", log_level="warning" app, port=config.server_config.port, host="0.0.0.0", log_level="warning"