Update app.py

app.py

@@ -100,7 +100,8 @@ def pics_pairwise_inference(background, img_a, mask_a, img_b, mask_b):
     ddim_sampler = DDIMSampler(model)
     
     # 唯一改动：对接 Gradio 的 PIL 转 OpenCV
-    back_image = cv2.cvtColor(np.array(background), cv2.COLOR_RGB2BGR)
+    # back_image = cv2.cvtColor(np.array(background), cv2.COLOR_RGB2BGR)
+    back_image = np.array(background)
     
     item_with_collage = {}
     objs = [(img_a, mask_a), (img_b, mask_b)]
@@ -155,11 +156,16 @@ def pics_pairwise_inference(background, img_a, mask_a, img_b, mask_b):
     # 但由于你 207 脚本最后写的是 cv2.imwrite(..., gen_image[:, :, ::-1])，
     # 说明在你的 crop_back 之后结果是 BGR。
     # 为了保险，我们在这里把模型结果转成 BGR 再给 crop_back
-    pred_bgr = cv2.cvtColor(pred_res, cv2.COLOR_RGB2BGR)
-    final_bgr = crop_back(pred_bgr, back_image, item_with_collage['extra_sizes'], 
-                          item_with_collage['hint_sizes0'], item_with_collage['hint_sizes1'], is_masked=True)
+    # pred_bgr = cv2.cvtColor(pred_res, cv2.COLOR_RGB2BGR)
+    # final_bgr = crop_back(pred_bgr, back_image, item_with_collage['extra_sizes'], 
+    #                       item_with_collage['hint_sizes0'], item_with_collage['hint_sizes1'], is_masked=True)
     
-    return cv2.cvtColor(final_bgr, cv2.COLOR_BGR2RGB)
+    # return cv2.cvtColor(final_bgr, cv2.COLOR_BGR2RGB)
+    final_image = crop_back(pred_res, back_image, item_with_collage['extra_sizes'], 
+                            item_with_collage['hint_sizes0'], item_with_collage['hint_sizes1'], is_masked=True)
+    
+    # 直接返回，因为 Gradio 也要 RGB
+    return final_image
 
 with gr.Blocks(title="PICS: Pairwise Spatial Compositing") as demo:
     gr.Markdown("# 🚀 PICS: Pairwise Image Compositing (5-Input Framework)")