From 410bef0853b61771729dd30f2f2df7a2906a1d4a Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 24 Sep 2022 09:16:54 +0300
Subject: [PATCH] do not hide preview image too early

---
 modules/ui.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/ui.py b/modules/ui.py
index 1c9b0f37b..e0ed26d95 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -197,7 +197,7 @@ def check_progress_call(id_part):
 
         image = shared.state.current_image
 
-        if image is None or progress >= 1:
+        if image is None:
             image = gr.update(value=None)
         else:
             preview_visibility = gr_show(True)