Sophie98 commited on
Commit
e4e8de7
β€’
1 Parent(s): 40845c9

fix errors

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -92,6 +92,8 @@ def style_sofa(input_img: np.ndarray, style_img: np.ndarray):
92
  # generate mask for image
93
  mask = get_mask(resized_img)
94
  styled_sofa = create_styledSofa(resized_img,resized_style)
 
 
95
  # postprocess the final image
96
  new_sofa = replace_sofa(resized_img,mask,styled_sofa)
97
  new_sofa = new_sofa.crop(box)
@@ -114,7 +116,7 @@ demo = gr.Interface(
114
  title="πŸ›‹ Style your sofa πŸ›‹ ",
115
  description="Customize your sofa to your wildest dreams!\
116
  \nProvide a picture of your sofa and a desired pattern\
117
- \n or choose one of the examples below",
118
  )
119
 
120
  if __name__ == "__main__":
 
92
  # generate mask for image
93
  mask = get_mask(resized_img)
94
  styled_sofa = create_styledSofa(resized_img,resized_style)
95
+ print(type(styled_sofa))
96
+ print(styled_sofa.shape)
97
  # postprocess the final image
98
  new_sofa = replace_sofa(resized_img,mask,styled_sofa)
99
  new_sofa = new_sofa.crop(box)
 
116
  title="πŸ›‹ Style your sofa πŸ›‹ ",
117
  description="Customize your sofa to your wildest dreams!\
118
  \nProvide a picture of your sofa and a desired pattern\
119
+ or choose one of the examples below",
120
  )
121
 
122
  if __name__ == "__main__":