Spaces:
Runtime error
Runtime error
ovshake
commited on
Commit
•
ae6d3bd
1
Parent(s):
6252c13
add mask pil
Browse files
app.py
CHANGED
|
@@ -142,6 +142,7 @@ stable_fashion_args.num_steps = num_steps
|
|
| 142 |
|
| 143 |
if file_name is not None:
|
| 144 |
result_image, mask_PIL = process_image(stable_fashion_args, inpainting_pipeline, net)
|
|
|
|
| 145 |
st.image(result_image, caption='Result')
|
| 146 |
st.image(mask_PIL, caption='Mask')
|
| 147 |
else:
|
|
|
|
| 142 |
|
| 143 |
if file_name is not None:
|
| 144 |
result_image, mask_PIL = process_image(stable_fashion_args, inpainting_pipeline, net)
|
| 145 |
+
print(np.unique(np.asarray(mask_PIL)))
|
| 146 |
st.image(result_image, caption='Result')
|
| 147 |
st.image(mask_PIL, caption='Mask')
|
| 148 |
else:
|