feat 模特预处理接口
fix
This commit is contained in:
@@ -3,8 +3,9 @@ import logging
|
||||
|
||||
from fastapi import APIRouter, HTTPException
|
||||
|
||||
from app.schemas.design import DesignModel, DesignProgressModel
|
||||
from app.schemas.design import DesignModel, DesignProgressModel, ModelProgressModel
|
||||
from app.schemas.response_template import ResponseModel
|
||||
from app.service.design.model_process_service import model_transpose
|
||||
from app.service.design.service import generate
|
||||
from app.service.design.utils.redis_utils import Redis
|
||||
|
||||
@@ -38,3 +39,16 @@ def get_progress(request_data: DesignProgressModel):
|
||||
logger.warning(f"get_progress Run Exception @@@@@@:{e}")
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
return ResponseModel(data=data)
|
||||
|
||||
|
||||
@router.post('/model_process')
|
||||
def model_process(request_data: ModelProgressModel):
|
||||
try:
|
||||
logger.info(f"model_process request item is : @@@@@@:{request_data.dict()}")
|
||||
|
||||
data = model_transpose(image_path=request_data.model_path)
|
||||
logger.info(f"model_process response @@@@@@:{json.dumps(data, indent=4)}")
|
||||
except Exception as e:
|
||||
logger.warning(f"model_process Run Exception @@@@@@:{e}")
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
return ResponseModel(data=data)
|
||||
|
||||
Reference in New Issue
Block a user