Update app.py
Browse files
app.py
CHANGED
|
@@ -36,6 +36,7 @@ def process_image(image, prompt):
|
|
| 36 |
|
| 37 |
# Convert image to BGR format
|
| 38 |
image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
|
|
|
|
| 39 |
# Convert mask to grayscale format
|
| 40 |
mask_image = cv2.cvtColor(np.array(mask_image), cv2.COLOR_RGB2GRAY)
|
| 41 |
|
|
|
|
| 36 |
|
| 37 |
# Convert image to BGR format
|
| 38 |
image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
|
| 39 |
+
imageio.imwrite("./data/data.png", image)
|
| 40 |
# Convert mask to grayscale format
|
| 41 |
mask_image = cv2.cvtColor(np.array(mask_image), cv2.COLOR_RGB2GRAY)
|
| 42 |
|