Update app.py

app.py

@@ -82,6 +82,7 @@ def process(init_image, mask):
     print(f'liuyz_1_here_')
     
     # image, alpha_channel = load_img(origin_image_bytes)
+    # Origin image shape: (512, 512, 3)
     original_shape = init_image.shape
     interpolation = cv2.INTER_CUBIC
     
@@ -189,6 +190,7 @@ def read_content(file_path: str) -> str:
 
 def predict(dict, prompt=""):
     init_image = dict["image"] # .convert("RGB") #.resize((512, 512))
+    print(init_image)
     mask = dict["mask"] # .convert("RGB") #.resize((512, 512))
     output = process(init_image, mask)
     # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)