diff --git a/nodes/ImageNode.py b/nodes/ImageNode.py index 96faf0e..a241904 100644 --- a/nodes/ImageNode.py +++ b/nodes/ImageNode.py @@ -1069,7 +1069,7 @@ class Image3D: mask=mask.convert('L') bg_image=None - if upload['bg_image']: + if 'bg_image' in upload: bg_image = base64_to_image(upload['bg_image']) bg_image=bg_image.convert('RGB') bg_image=pil2tensor(bg_image)