Update app.py

app.py

@@ -110,12 +110,12 @@ def predict(
         input_image = input_image.convert("RGB").resize((int(size1 / size2 * 1024), 1024))
 
     img = np.array(input_image.convert("RGB"))
-
-    W = int(np.shape(img)[1] - np.shape(img)[1] % 16)
+    
     H = int(np.shape(img)[0] - np.shape(img)[0] % 16)
-
-    input_image = input_image.resize((H, W))
-    uploaded_mask = uploaded_mask.resize((H, W))
+    W = int(np.shape(img)[1] - np.shape(img)[1] % 16)
+    
+    input_image = input_image.resize((W, H))
+    uploaded_mask = uploaded_mask.resize((W, H))
 
     if seed == -1:
         seed = random.randint(1, 2147483647)