update server_fastapi.py: 缺少web文件时添加提示信息。configyml添加更多帮助注释

This commit is contained in:
Sora
2023-12-01 10:13:42 +08:00
parent d4a2bc225e
commit 95f44ed4fe
2 changed files with 19 additions and 9 deletions

View File

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

View File

@@ -149,17 +149,24 @@ if __name__ == "__main__":
app = FastAPI()
app.logger = logger
# 挂载静态文件
logger.info("开始挂载网页页面")
StaticDir: str = "./Web"
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()]
for dirName in dirs:
app.mount(
f"/{dirName}",
StaticFiles(directory=f"./{StaticDir}/{dirName}"),
name=dirName,
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()]
files = [fir.name for fir in os.scandir(StaticDir) if fir.is_dir()]
for dirName in dirs:
app.mount(
f"/{dirName}",
StaticFiles(directory=f"./{StaticDir}/{dirName}"),
name=dirName,
)
loaded_models = Models()
# 加载模型
logger.info("开始加载模型")
models_info = config.server_config.models
for model_info in models_info:
loaded_models.init_model(
@@ -605,7 +612,8 @@ if __name__ == "__main__":
logger.warning("本地服务,请勿将服务端口暴露于外网")
logger.info(f"api文档地址 http://127.0.0.1:{config.server_config.port}/docs")
webbrowser.open(f"http://127.0.0.1:{config.server_config.port}")
if os.path.isdir(StaticDir):
webbrowser.open(f"http://127.0.0.1:{config.server_config.port}")
uvicorn.run(
app, port=config.server_config.port, host="0.0.0.0", log_level="warning"
)