huzey commited on
Commit
bd4e31e
·
1 Parent(s): 04d4e3a

update blend

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4504,7 +4504,7 @@ with demo:
4504
  for i, ncut_image in enumerate(ncut_images):
4505
  original_image = original_images[i]
4506
  original_image = np.array(original_image)
4507
- if original_image.dim == 2:
4508
  original_image = np.stack([original_image] * 3, axis=-1)
4509
  ncut_image = ncut_image.resize(original_image.shape[:2][::-1], Image.LANCZOS)
4510
  ncut_image = np.array(ncut_image)
 
4504
  for i, ncut_image in enumerate(ncut_images):
4505
  original_image = original_images[i]
4506
  original_image = np.array(original_image)
4507
+ if len(original_image.shape) == 2:
4508
  original_image = np.stack([original_image] * 3, axis=-1)
4509
  ncut_image = ncut_image.resize(original_image.shape[:2][::-1], Image.LANCZOS)
4510
  ncut_image = np.array(ncut_image)