Update app.py

app.py

@@ -159,13 +159,14 @@ def model_process(image, mask):
         config.sd_seed = random.randint(1, 999999999)
     
     # logger.info(f"Origin image shape: {original_shape}")
-    print(f"Origin image shape: {original_shape}")
+    print(f"Origin image shape: {original_shape} / {type(image)}")
     image = resize_max_size(image, size_limit=size_limit, interpolation=interpolation)
     # logger.info(f"Resized image shape: {image.shape}")
-    print(f"Resized image shape: {image.shape}")
+    print(f"Resized image shape: {image.shape} / {type(image)}")
     
     #mask, _ = load_img(input["mask"].read(), gray=True)
     mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
+    print(f"mask image shape: {mask.shape} / {type(mask)}")
 
     if model is None:
         return None
@@ -222,15 +223,19 @@ def read_content(file_path: str) -> str:
 def predict(dict):
     # print(f'liuyz_0_', dict)  
      
+    '''
     image = dict["image"] # .convert("RGB") #.resize((512, 512))
     # target_size = (init_image.shape[0], init_image.shape[1])
     print(f'liuyz_1_', image.shape)
     print(f'liuyz_2_', image.convert("RGB").shape)
     print(f'liuyz_3_', image.convert("RGB").resize((512, 512)).shape)
     # mask = dict["mask"] # .convert("RGB") #.resize((512, 512))
+    '''
     
     image = Image.fromarray(dict["image"])
-    mask = Image.fromarray(dict["mask"])
+    mask = Image.fromarray(dict["mask"]).convert("RGB")
+    print(f'mask___1 = {mask.shape}')
+    
     output = model_process(dict["image"], dict["mask"])
     # output = mask #output.images[0]
     # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)