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__":
|