Fix image padding.

app.py

@@ -23,17 +23,19 @@ def denormalize_and_clip_as_numpy (im: Tensor) -> np.ndarray:
     im = im.squeeze(0)
     return np.ascontiguousarray(denormalize_and_clip_as_tensor(im).permute(1,2,0).detach().cpu().numpy())
 
+def pad_width (size: int, multiple: int):
+    return 0 if size % multiple == 0 else multiple - (size%multiple)
+
 # pad image to a multiple of 64
 def pad_image(im, multiple=64):
     # B,C,H,W
     rows = im.shape[2]
     cols = im.shape[3]
-    if rows % multiple == 0 and cols % multiple == 0:
+    rows_to_pad = pad_width(rows, multiple)
+    cols_to_pad = pad_width(cols, multiple)
+    if rows_to_pad == 0 and cols_to_pad == 0:
         return im
-    else:
-        rows_to_pad = multiple - (rows % multiple)
-        cols_to_pad = multiple - (cols % multiple)
-        return transforms.Pad(padding=(0, 0, cols_to_pad, rows_to_pad), padding_mode='reflect')(im)
+    return transforms.Pad(padding=(0, 0, cols_to_pad, rows_to_pad), padding_mode='reflect')(im)
 
 def undo_antialiasing(im):
     im_torch = torch.from_numpy (im).permute(2,0,1).unsqueeze(0).float() / 255.0