fix:启动端口改由环境变量控制

This commit is contained in:
zcr
2026-01-27 10:40:50 +08:00
parent 1bbb9c945e
commit 43fd576da6
2 changed files with 7 additions and 1 deletions

View File

@@ -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表示生产环境运行")

View File

@@ -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 = {