diff options
Diffstat (limited to 'modules/extras.py')
| -rw-r--r-- | modules/extras.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/extras.py b/modules/extras.py index 15873204..af6e631f 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -39,6 +39,8 @@ def run_extras(extras_mode, image, image_folder, gfpgan_visibility, codeformer_v outputs = []
for image, image_name in zip(imageArr, imageNameArr):
+ if image is None:
+ return outputs, "Please select an input image.", ''
existing_pnginfo = image.info or {}
image = image.convert("RGB")
|