fixing error?

app.py

@@ -22,6 +22,7 @@ def style_sofa(input_img: np.ndarray, style_img: np.ndarray):
     # generate mask for image
     mask = get_mask(resized_img)
     styled_sofa = create_styledSofa(resized_img,resized_style)
+    print('Created styled image')
     new_sofa = replace_sofa(resized_img,mask,styled_sofa)
     return new_sofa
 

segmentation.py

@@ -51,14 +51,18 @@ def replace_sofa(image,mask,styled_sofa):
     # print(mask.shape)
     # mask = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)
     # print(mask.shape)
+    print('started function')
     image = np.array(image)
     #image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     styled_sofa = cv2.cvtColor(styled_sofa, cv2.COLOR_BGR2RGB)
 
+    print('reading masks')
     _, mask = cv2.threshold(mask, 10, 255, cv2.THRESH_BINARY)
     mask_inv = cv2.bitwise_not(mask)
+    print('creating parts')
     image_bg = cv2.bitwise_and(image,image,mask = mask_inv)
     sofa_fg = cv2.bitwise_and(styled_sofa,styled_sofa,mask = mask)
+    print('combining parts')
     new_image = cv2.add(image_bg,sofa_fg)
     return new_image
 

styleTransfer.py

@@ -57,8 +57,6 @@ def create_styledSofa(sofa,style):
     path_sofa,path_style = 'sofa.jpg','style.jpg'
     sofa.save(path_sofa)
     style.save(path_style)
-    #newpath_sofa = resize_sofa(path_sofa)
-    #newpath_style = resize_style(path_style)
     os.system("python3 test.py  --content "+path_sofa+" \
                     --style "+path_style+" \
                     --output . \