Spaces:
Running
Running
Lodor
commited on
Commit
•
4009180
1
Parent(s):
ff029de
Fix canvas error
Browse files- app.py +3 -3
- docker-compose.yml +3 -3
app.py
CHANGED
@@ -66,7 +66,7 @@ if uploaded_file is not None:
|
|
66 |
|
67 |
stroke_width = st.slider("Brush size", 1, 100, 50)
|
68 |
|
69 |
-
st.write("Now draw (brush) the part of image that you want to remove
|
70 |
|
71 |
# Canvas size logic
|
72 |
canvas_bg = deepcopy(img_input)
|
@@ -88,7 +88,7 @@ if uploaded_file is not None:
|
|
88 |
)
|
89 |
|
90 |
if canvas_result.image_data is not None:
|
91 |
-
im = np.array(Image.fromarray(canvas_result.image_data).resize(img_input.size))
|
92 |
background = np.where(
|
93 |
(im[:, :, 0] == 0) &
|
94 |
(im[:, :, 1] == 0) &
|
@@ -100,7 +100,7 @@ if uploaded_file is not None:
|
|
100 |
(im[:, :, 2] == 255)
|
101 |
)
|
102 |
im[background]=[0,0,0,255]
|
103 |
-
im[drawing]=[0,0,0,0] #RGBA
|
104 |
|
105 |
reuse = False
|
106 |
|
|
|
66 |
|
67 |
stroke_width = st.slider("Brush size", 1, 100, 50)
|
68 |
|
69 |
+
st.write("**Now draw (brush) the part of image that you want to remove.**")
|
70 |
|
71 |
# Canvas size logic
|
72 |
canvas_bg = deepcopy(img_input)
|
|
|
88 |
)
|
89 |
|
90 |
if canvas_result.image_data is not None:
|
91 |
+
im = np.array(Image.fromarray(canvas_result.image_data.astype(np.uint8)).resize(img_input.size))
|
92 |
background = np.where(
|
93 |
(im[:, :, 0] == 0) &
|
94 |
(im[:, :, 1] == 0) &
|
|
|
100 |
(im[:, :, 2] == 255)
|
101 |
)
|
102 |
im[background]=[0,0,0,255]
|
103 |
+
im[drawing]=[0,0,0,0] # RGBA
|
104 |
|
105 |
reuse = False
|
106 |
|
docker-compose.yml
CHANGED
@@ -1,12 +1,12 @@
|
|
1 |
---
|
2 |
version: '3'
|
3 |
services:
|
4 |
-
|
5 |
build: .
|
6 |
-
container_name:
|
7 |
restart: unless-stopped
|
8 |
ports:
|
9 |
-
-
|
10 |
volumes:
|
11 |
- .:/app
|
12 |
environment:
|
|
|
1 |
---
|
2 |
version: '3'
|
3 |
services:
|
4 |
+
sthf-remove-photo-object:
|
5 |
build: .
|
6 |
+
container_name: sthf-remove-photo-object
|
7 |
restart: unless-stopped
|
8 |
ports:
|
9 |
+
- 41003:8501
|
10 |
volumes:
|
11 |
- .:/app
|
12 |
environment:
|