fix:启动端口改由环境变量控制
This commit is contained in:
@@ -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表示生产环境运行")
|
||||
|
||||
|
||||
@@ -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 = {
|
||||
|
||||
Reference in New Issue
Block a user