try to fix error

app.py

@@ -96,15 +96,15 @@ def style_sofa(input_img: np.ndarray, style_img: np.ndarray):
     input_img,style_img = Image.fromarray(input_img),Image.fromarray(style_img)
     resized_img,box = resize_sofa(input_img)
     resized_style = resize_style(style_img)
-    resized_style.save('resized_style.jpg')
+    #resized_style.save('resized_style.jpg')
     # generate mask for image
     print('generating mask...')
     mask = get_mask(resized_img)
-    mask.save('mask.jpg')
+    #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')
+    #styled_sofa.save('styled_sofa.jpg')
     # postprocess the final image
     print('Replacing sofa...')
     new_sofa = replace_sofa(resized_img,mask,styled_sofa)

styleTransfer.py

@@ -34,7 +34,6 @@ def content_transform():
     transform = transforms.Compose(transform_list)
     return transform
 
-
 def StyleTransformer(content_img: Image, style_img: Image):
 
     # Advanced options