diff --git a/app/api/api_generate_image.py b/app/api/api_generate_image.py
index 41cf989..cce1300 100644
--- a/app/api/api_generate_image.py
+++ b/app/api/api_generate_image.py
@@ -3,7 +3,7 @@ import logging
 
 from fastapi import APIRouter, BackgroundTasks, HTTPException
 
-from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel
+from app.schemas.generate_image import GenerateImageModel, GenerateProductImageModel, GenerateSingleLogoImageModel, GenerateRelightImageModel
 from app.schemas.response_template import ResponseModel
 from app.service.generate_image.service_generate_image import GenerateImage, infer_cancel as generate_image_infer_cancel
 from app.service.generate_image.service_generate_product_image import GenerateProductImage, infer_cancel as generate_product_image_cancel
@@ -145,7 +145,7 @@ def generate_product_image(tasks_id: str):
 
 
 @router.post("/generate_relight_image")
-def generate_relight_image(request_item: GenerateProductImageModel, background_tasks: BackgroundTasks):
+def generate_relight_image(request_item: GenerateRelightImageModel, background_tasks: BackgroundTasks):
     """
         创建一个具有以下参数的请求体:
         - **tasks_id**: 任务id 用于取消生成任务和获取生成结果
diff --git a/app/service/design/items/pipelines/painting.py b/app/service/design/items/pipelines/painting.py
index a738455..424a395 100644
--- a/app/service/design/items/pipelines/painting.py
+++ b/app/service/design/items/pipelines/painting.py
@@ -118,6 +118,7 @@ class PrintPainting(object):
 
                         print_background = cv2.cvtColor(np.array(source_image_pil), cv2.COLOR_RGBA2BGR)
                         mask_background = cv2.cvtColor(np.array(source_image_pil_mask), cv2.COLOR_RGBA2BGR)
+                        ret, mask_background = cv2.threshold(mask_background, 124, 255, cv2.THRESH_BINARY)
                     else:
                         mask = self.get_mask_inv(image)
                         mask = np.expand_dims(mask, axis=2)