do not hide preview image too early

3 years ago · 410bef0853
parent 31ac0baba0
commit 410bef0853
1 changed files with 1 additions and 1 deletions
--- 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)