still testing

app.py

@@ -98,8 +98,8 @@ def style_sofa(input_img: np.ndarray, style_img: np.ndarray):
     resized_style = resize_style(style_img)
     #resized_style.save('resized_style.jpg')
     # generate mask for image
-    print('generating mask...')
-    mask = get_mask(resized_img)
+    # print('generating mask...')
+    # mask = get_mask(resized_img)
     #mask.save('mask.jpg')
     # Created a styled sofa
     # print('Styling sofa...')
@@ -110,7 +110,7 @@ def style_sofa(input_img: np.ndarray, style_img: np.ndarray):
     # new_sofa = replace_sofa(resized_img,mask,styled_sofa)
     # new_sofa = new_sofa.crop(box)
     print('Finishing job', id)
-    return mask
+    return resized_img
 
 demo = gr.Interface(
     style_sofa,