Update app.py
Browse files
app.py
CHANGED
|
@@ -192,9 +192,10 @@ if __name__ == "__main__":
|
|
| 192 |
mask_cv = 255-mask_cv
|
| 193 |
|
| 194 |
img_C = cv2.seamlessClone(dict, img_cv, mask_cv, (192, 256), cv2.NORMAL_CLONE)
|
|
|
|
| 195 |
img_C = cv2.cvtColor(img_C, cv2.COLOR_BGR2RGB)
|
| 196 |
-
img_C =
|
| 197 |
-
|
| 198 |
return img_C, mask_gray
|
| 199 |
|
| 200 |
|
|
|
|
| 192 |
mask_cv = 255-mask_cv
|
| 193 |
|
| 194 |
img_C = cv2.seamlessClone(dict, img_cv, mask_cv, (192, 256), cv2.NORMAL_CLONE)
|
| 195 |
+
img_C = img_C.astype(np.uint8)
|
| 196 |
img_C = cv2.cvtColor(img_C, cv2.COLOR_BGR2RGB)
|
| 197 |
+
img_C = to_pil_image(img_C)
|
| 198 |
+
|
| 199 |
return img_C, mask_gray
|
| 200 |
|
| 201 |
|