testing

app.py

@@ -102,15 +102,15 @@ def style_sofa(input_img: np.ndarray, style_img: np.ndarray):
     mask = get_mask(resized_img)
     #mask.save('mask.jpg')
     # Created a styled sofa
-    print('Styling sofa...')
-    styled_sofa = create_styledSofa(resized_img,resized_style)
-    #styled_sofa.save('styled_sofa.jpg')
-    # postprocess the final image
-    print('Replacing sofa...')
-    new_sofa = replace_sofa(resized_img,mask,styled_sofa)
-    new_sofa = new_sofa.crop(box)
+    # print('Styling sofa...')
+    # styled_sofa = create_styledSofa(resized_img,resized_style)
+    # #styled_sofa.save('styled_sofa.jpg')
+    # # postprocess the final image
+    # print('Replacing sofa...')
+    # new_sofa = replace_sofa(resized_img,mask,styled_sofa)
+    # new_sofa = new_sofa.crop(box)
     print('Finishing job', id)
-    return new_sofa
+    return mask
 
 demo = gr.Interface(
     style_sofa,