From 740070ea9cdb254209f66417418f2a4af8b099d6 Mon Sep 17 00:00:00 2001
From: d8ahazard <d8ahazard@gmail.com>
Date: Mon, 26 Sep 2022 09:29:50 -0500
Subject: Re-implement universal model loading

---
 modules/extras.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'modules/extras.py')

diff --git a/modules/extras.py b/modules/extras.py
index 382ffa7d..4c95cf76 100644
--- a/modules/extras.py
+++ b/modules/extras.py
@@ -36,6 +36,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")
-- 
cgit v1.2.1