Update app.py

app.py

@@ -76,10 +76,14 @@ model = None
 def model_process(image, mask):
     global model
     
+    if mask.shape[0] == image.shape[1] and mask.shape[1] == image.shape[0]:
+        # rotate image
+        image = np.transpose(image[::-1, ...][:, ::-1], axes=(1, 0, 2))[::-1, ...]
+        
     original_shape = image.shape
     interpolation = cv2.INTER_CUBIC
     
-    size_limit = 512 #1080 # "Original"
+    size_limit = 1080 #1080 # "Original"
     if size_limit == "Original":
         size_limit = max(image.shape)
     else:
@@ -112,13 +116,13 @@ def model_process(image, mask):
     if config.sd_seed == -1:
         config.sd_seed = random.randint(1, 999999999)
 
-    print(f"Origin image shape_0_: {original_shape} / {image[250][250]} / {size_limit}")
+    print(f"Origin image shape_0_: {original_shape} / {size_limit}")
     image = resize_max_size(image, size_limit=size_limit, interpolation=interpolation)
-    print(f"Resized image shape_1_: {image.shape} / {image[250][250]}")
+    print(f"Resized image shape_1_: {image.shape}")
     
-    print(f"mask image shape_0_: {mask.shape} / {type(mask)} / {mask[250][250]}")
+    print(f"mask image shape_0_: {mask.shape} / {type(mask)}")
     mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
-    print(f"mask image shape_1_: {mask.shape} / {type(mask)} / {mask[250][250]}")
+    print(f"mask image shape_1_: {mask.shape} / {type(mask)}")
 
     if model is None:
         return None