Update

model.py

@@ -120,8 +120,7 @@ class Model:
         images = [np.asarray(image) for image in images]
         nrows = (len(images) + ncols - 1) // ncols
         h, w = images[0].shape[:2]
-        d = nrows * ncols - len(images)
-        if d > 0:
+        if (d := nrows * ncols - len(images)) > 0:
             images += [np.full((h, w, 3), 255, dtype=np.uint8)] * d
         grid = np.asarray(images).reshape(nrows, ncols, h, w, 3).transpose(
             0, 2, 1, 3, 4).reshape(nrows * h, ncols * w, 3)