diff --git a/app/config.py b/app/config.py index 7718fca..d2206de 100644 --- a/app/config.py +++ b/app/config.py @@ -18,6 +18,10 @@ class Settings(BaseSettings): ) # 启动端口 SERVE_PORT: int = Field(default=8000, description='') + + # 换脸模型服务 + RE_FACE_MODEL_URL: str = Field(default="10.1.1.240:10071", description='') + # 调试配饰 LOCAL: int = Field(default=0, description="是否在本地运行,1表示本地运行,0表示生产环境运行") diff --git a/app/server/ReFace/server.py b/app/server/ReFace/server.py index 413e5f1..f8c18ad 100644 --- a/app/server/ReFace/server.py +++ b/app/server/ReFace/server.py @@ -4,6 +4,8 @@ import litserve as ls import requests from pydantic import BaseModel +from app.config import settings + class PredictRequest(BaseModel): input_image_list: list[str] # 待换脸图片 @@ -17,7 +19,7 @@ class ReFace(ls.LitAPI): def predict(self, request): # 服务的 URL - url = "http://10.1.1.240:10071/predict" + url = f"http://{settings.RE_FACE_MODEL_URL}/predict" # 请求头 headers = {