Update app.py

app.py

@@ -76,7 +76,7 @@ model = None
 def model_process(image, mask):
     global model
     
-    if mask.shape[0] == image.shape[1] and mask.shape[1] == image.shape[0]:
+    if mask.shape[0] == image.shape[1] and mask.shape[1] == image.shape[0] and mask.shape[0] != mask.shape[1]:
         # rotate image
         image = np.transpose(image[::-1, ...][:, ::-1], axes=(1, 0, 2))[::-1, ...]