basso4 commited on
Commit
c09dc2b
Β·
verified Β·
1 Parent(s): 26167ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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 = Image.fromarray(img_C, mode="RGB")
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