Spaces:
Build error
Build error
Sophie98
commited on
Commit
β’
e4e8de7
1
Parent(s):
40845c9
fix errors
Browse files
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 |
-
|
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__":
|