删除outfit 图片item文字信息,新增换脸模型接口。
This commit is contained in:
59
app/server/ReFace/server.py
Normal file
59
app/server/ReFace/server.py
Normal file
@@ -0,0 +1,59 @@
|
||||
import json
|
||||
|
||||
import litserve as ls
|
||||
import requests
|
||||
from pydantic import BaseModel
|
||||
|
||||
|
||||
class PredictRequest(BaseModel):
|
||||
input_image_list: list[str] # 待换脸图片
|
||||
input_face: str # 目标脸图片
|
||||
threshold: float = 0.2 # 相似度 max:0.5
|
||||
|
||||
|
||||
class ReFace(ls.LitAPI):
|
||||
def decode_request(self, request: PredictRequest):
|
||||
return request
|
||||
|
||||
def predict(self, request):
|
||||
# 服务的 URL
|
||||
url = "http://10.1.1.240:10071/predict"
|
||||
|
||||
# 请求头
|
||||
headers = {
|
||||
"accept": "application/json",
|
||||
"Content-Type": "application/json"
|
||||
}
|
||||
|
||||
# 请求体数据
|
||||
# 这里的结构要和你的 LitServe 服务的 LitAPI.decode_request 预期的一致
|
||||
data = {
|
||||
"input_image_list": request.input_image_list,
|
||||
"input_face": request.input_face,
|
||||
"threshold": request.threshold
|
||||
}
|
||||
|
||||
try:
|
||||
# 使用 requests.post 发送请求
|
||||
# 使用 json= 参数可以自动将 Python 字典转换为 JSON 格式,并设置 Content-Type 头部
|
||||
response = requests.post(url, headers=headers, json=data)
|
||||
|
||||
# 检查 HTTP 响应状态码,如果不是 200/201 等成功状态,将抛出异常
|
||||
response.raise_for_status()
|
||||
|
||||
# 打印返回的 JSON 结果
|
||||
print("成功调用 LitServe 接口,返回结果:")
|
||||
# .json() 方法将响应体解析为 Python 字典
|
||||
print(json.dumps(response.json(), indent=4, ensure_ascii=False))
|
||||
return response.json()
|
||||
|
||||
except requests.exceptions.RequestException as e:
|
||||
# 处理请求失败、连接错误或 HTTP 错误状态
|
||||
print(f"请求发生错误: {e}")
|
||||
if 'response' in locals() and response is not None:
|
||||
print(f"响应状态码: {response.status_code}")
|
||||
try:
|
||||
# 尝试打印服务器返回的错误详情
|
||||
print(f"服务器错误详情: {response.json()}")
|
||||
except:
|
||||
print(f"服务器错误详情 (非 JSON 格式): {response.text}")
|
||||
Reference in New Issue
Block a user