Sophie98 commited on
Commit
b5a3fed
·
1 Parent(s): 8683396

fix error again?

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def style_sofa(input_img: np.ndarray, style_img: np.ndarray):
27
  styled_sofa = create_styledSofa(resized_img,resized_style)
28
  print('resizing')
29
  new_sofa = replace_sofa(resized_img,mask,styled_sofa)
30
- new_sofa = cv2.cvtColor(Image.fromarray(new_sofa), cv2.COLOR_BGR2RGB).crop(box)
31
  return new_sofa
32
 
33
  image = gr.inputs.Image()
 
27
  styled_sofa = create_styledSofa(resized_img,resized_style)
28
  print('resizing')
29
  new_sofa = replace_sofa(resized_img,mask,styled_sofa)
30
+ new_sofa = Image.fromarray(new_sofa).crop(box)
31
  return new_sofa
32
 
33
  image = gr.inputs.Image()