fix:启动端口改由环境变量控制
This commit is contained in:
@@ -18,6 +18,10 @@ class Settings(BaseSettings):
|
|||||||
)
|
)
|
||||||
# 启动端口
|
# 启动端口
|
||||||
SERVE_PORT: int = Field(default=8000, description='')
|
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表示生产环境运行")
|
LOCAL: int = Field(default=0, description="是否在本地运行,1表示本地运行,0表示生产环境运行")
|
||||||
|
|
||||||
|
|||||||
@@ -4,6 +4,8 @@ import litserve as ls
|
|||||||
import requests
|
import requests
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
from app.config import settings
|
||||||
|
|
||||||
|
|
||||||
class PredictRequest(BaseModel):
|
class PredictRequest(BaseModel):
|
||||||
input_image_list: list[str] # 待换脸图片
|
input_image_list: list[str] # 待换脸图片
|
||||||
@@ -17,7 +19,7 @@ class ReFace(ls.LitAPI):
|
|||||||
|
|
||||||
def predict(self, request):
|
def predict(self, request):
|
||||||
# 服务的 URL
|
# 服务的 URL
|
||||||
url = "http://10.1.1.240:10071/predict"
|
url = f"http://{settings.RE_FACE_MODEL_URL}/predict"
|
||||||
|
|
||||||
# 请求头
|
# 请求头
|
||||||
headers = {
|
headers = {
|
||||||
|
|||||||
Reference in New Issue
Block a user