Update app.py

app.py

@@ -190,11 +190,13 @@ def pics_pairwise_inference(background, img_a, mask_a, img_b, mask_b):
         
         x_samples = model.decode_first_stage(samples)
         x_samples = (einops.rearrange(x_samples, 'b c h w -> b h w c') * 127.5 + 127.5).cpu().numpy()
-        pred = np.clip(x_samples[0], 0, 255).astype(np.uint8)
+        pred_rgb = np.clip(x_samples[0], 0, 255).astype(np.uint8)
+
+        pred_bgr = cv2.cvtColor(pred_rgb, cv2.COLOR_RGB2BGR)
         
     # 结果后处理
     side = max(back_image.shape[0], back_image.shape[1])
-    pred = cv2.resize(pred, (side, side))
+    pred = cv2.resize(pred_bgr, (side, side))
     pred = crop_back(pred, back_image, item_with_collage['extra_sizes'], 
                      item_with_collage['hint_sizes0'], item_with_collage['hint_sizes1'], is_masked=True)