update server_fastapi.py: 缺少web文件时添加提示信息。configyml添加更多帮助注释
This commit is contained in:
@@ -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: ""
|
||||||
|
|||||||
@@ -149,17 +149,24 @@ if __name__ == "__main__":
|
|||||||
app = FastAPI()
|
app = FastAPI()
|
||||||
app.logger = logger
|
app.logger = logger
|
||||||
# 挂载静态文件
|
# 挂载静态文件
|
||||||
|
logger.info("开始挂载网页页面")
|
||||||
StaticDir: str = "./Web"
|
StaticDir: str = "./Web"
|
||||||
dirs = [fir.name for fir in os.scandir(StaticDir) if fir.is_dir()]
|
if not os.path.isdir(StaticDir):
|
||||||
files = [fir.name for fir in os.scandir(StaticDir) if fir.is_dir()]
|
logger.warning(
|
||||||
for dirName in dirs:
|
"缺少网页资源,无法开启网页页面,如有需要请在 https://github.com/jiangyuxiaoxiao/Bert-VITS2-UI 或者Bert-VITS对应版本的release页面下载"
|
||||||
app.mount(
|
|
||||||
f"/{dirName}",
|
|
||||||
StaticFiles(directory=f"./{StaticDir}/{dirName}"),
|
|
||||||
name=dirName,
|
|
||||||
)
|
)
|
||||||
|
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()
|
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,7 +612,8 @@ 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")
|
||||||
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(
|
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"
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user