Update app.py

app.py

@@ -112,7 +112,7 @@ def add_gallery(image, model_str, gallery):
     if gallery is None: gallery = []
     with lock:
         if image is not None: gallery.insert(0, (image, model_str))
-        if image is not None: imgageHasUpdated(model_str)
+        # if image is not None: imgageHasUpdated(model_str)
     return gallery
 
 JS="""
@@ -218,6 +218,7 @@ with gr.Blocks(head=CSS + JS) as demo:
         with gr.Group():
             with gr.Row():
                 output = [gr.Image(label=m, show_download_button=True, elem_classes=["image-monitor"], interactive=False, width=112, height=112, show_share_button=False, format="png", visible=True) for m in default_models]
+                imgageHasUpdated(output)
                 current_models = [gr.Textbox(m, visible=False) for m in default_models]
 
     with gr.Column(scale=2):