Update app.py

app.py

@@ -182,8 +182,8 @@ if __name__ == "__main__":
                 dict = cv2.cvtColor(dict, cv2.COLOR_RGB2BGR)
 
                 #img_output
-                img_cv = rearrange(x_checked_image_torch[0], 'c h w -> h w c').cpu().numpy()
-                img_cv = (img_cv * 255).astype(np.uint8)
+                img_cv = rearrange(x_checked_image_torch[0], 'c h w -> h w c').cpu().numpy().astype(np.uint8)
+                img_cv = (img_cv * 255)
                 img_cv = cv2.cvtColor(img_cv, cv2.COLOR_RGB2BGR)
                 
                 #mask
@@ -196,7 +196,7 @@ if __name__ == "__main__":
                 
                 img_C = cv2.seamlessClone(dict, img_cv, mask_cv, (192, 256), cv2.NORMAL_CLONE)
                 img_C = cv2.cvtColor(img_C, cv2.COLOR_BGR2RGB)
-                img_C = Image.fromarray(img_C)
+                img_C = to_pil_image(img_C)
                 
                 return img_C, mask_gray