update server_fastapi.py: 缺少web文件时添加提示信息。configyml添加更多帮助注释
This commit is contained in:
@@ -121,8 +121,10 @@ server:
|
||||
port: 5000
|
||||
# 模型默认使用设备:但是当前并没有实现这个配置。
|
||||
device: "cuda"
|
||||
# 需要加载的所有模型的配置
|
||||
# 需要加载的所有模型的配置,可以填多个模型,也可以不填模型,等网页成功后手动加载模型
|
||||
# 不加载模型的配置格式:删除默认给的两个模型配置,给models赋值 [ ],也就是空列表。参考模型2的speakers 即 models: [ ]
|
||||
# 注意,所有模型都必须正确配置model与config的路径,空路径会导致加载错误。
|
||||
# 也可以不填模型,等网页加载成功后手动填写models。
|
||||
models:
|
||||
- # 模型的路径
|
||||
model: ""
|
||||
|
||||
@@ -149,7 +149,13 @@ if __name__ == "__main__":
|
||||
app = FastAPI()
|
||||
app.logger = logger
|
||||
# 挂载静态文件
|
||||
logger.info("开始挂载网页页面")
|
||||
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()]
|
||||
files = [fir.name for fir in os.scandir(StaticDir) if fir.is_dir()]
|
||||
for dirName in dirs:
|
||||
@@ -160,6 +166,7 @@ if __name__ == "__main__":
|
||||
)
|
||||
loaded_models = Models()
|
||||
# 加载模型
|
||||
logger.info("开始加载模型")
|
||||
models_info = config.server_config.models
|
||||
for model_info in models_info:
|
||||
loaded_models.init_model(
|
||||
@@ -605,6 +612,7 @@ if __name__ == "__main__":
|
||||
|
||||
logger.warning("本地服务,请勿将服务端口暴露于外网")
|
||||
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}")
|
||||
uvicorn.run(
|
||||
app, port=config.server_config.port, host="0.0.0.0", log_level="warning"
|
||||
|
||||
Reference in New Issue
Block a user