Upload app.py

app.py

@@ -672,7 +672,7 @@ class GuiSD:
 ## BEGIN MOD
                 img = save_images(img, metadata)
                 if not display_images:
-                    img = gr.update(value=img) if img else gr.update()
+                    img = img if img else gr.update()
 ## END MOD
                 info_state = "COMPLETE"