Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -136,7 +136,7 @@ def floorplan_generation(outline, num_of_rooms):
|
|
136 |
image_arr = np.array(image)
|
137 |
mask_bgr = cv2.cvtColor(image_arr, cv2.COLOR_RGB2BGR)
|
138 |
mask_gray = cv2.cvtColor(mask_bgr, cv2.COLOR_RGB2GRAY)
|
139 |
-
mask_white = cv2.threshold(mask_gray,
|
140 |
image_arr[mask_white<250]=(255,255,255)
|
141 |
image_arr_copy = image_arr.copy()
|
142 |
|
|
|
136 |
image_arr = np.array(image)
|
137 |
mask_bgr = cv2.cvtColor(image_arr, cv2.COLOR_RGB2BGR)
|
138 |
mask_gray = cv2.cvtColor(mask_bgr, cv2.COLOR_RGB2GRAY)
|
139 |
+
mask_white = cv2.threshold(mask_gray, 200, 255, cv2.THRESH_BINARY_INV)[1]
|
140 |
image_arr[mask_white<250]=(255,255,255)
|
141 |
image_arr_copy = image_arr.copy()
|
142 |
|