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

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