fix colors

app.py

@@ -71,7 +71,7 @@ def random_positioning(input, output_size=(1024, 1024)):
     
     resized_image = cv2.resize(input, new_size, interpolation=cv2.INTER_AREA)
     
-    background = np.zeros((output_size[1], output_size[0], 3), dtype=np.uint8)
+    background = np.ones((output_size[1], output_size[0], 3), dtype=np.uint8) * 255
     
     x_offset = random.randint(0, output_size[0] - new_size[0])
     y_offset = random.randint(0, output_size[1] - new_size[1])
@@ -126,6 +126,9 @@ def execute(image, prompt):
     cv2.imwrite("mask.jpg", maskHead('base_image.jpg'))
     
     im = Image.open('base_image.jpg')
+    np_arr = np.array(im)
+    rgb_image = cv2.cvtColor(np_image, cv2.COLOR_BGR2RGB)
+    im =  Image.fromarray(rgb_image)
     mask = Image.open('mask.jpg')
 
     result = pipe(