Update app.py

app.py

@@ -111,8 +111,8 @@ def predict(
 
     img = np.array(input_image.convert("RGB"))
 
-    W = int(np.shape(img)[0] - np.shape(img)[0] % 16)
-    H = int(np.shape(img)[1] - np.shape(img)[1] % 16)
+    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))
@@ -128,8 +128,8 @@ def predict(
         prompt=prompt,
         control_image=input_image.convert("RGB"),
         control_mask=uploaded_mask.convert("RGB"),
-        width=H,
-        height=W,
+        width=W,
+        height=H,
         num_inference_steps=ddim_steps,
         generator=torch.Generator("cuda").manual_seed(seed),
         guidance_scale=scale,