Update app.py

app.py

@@ -90,6 +90,7 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
         with lock:
             png_path =  model_str.replace("/", "_") + " - " + get_current_time() + "_" + str(theSeed) + ".png"
             # lnk_output(value=png_path)
+            imgageHasUpdated(png_path)
             image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, theSeed)
         return image
     return None
@@ -217,7 +218,6 @@ 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]
-                output.change(imgageHasUpdated)
                 current_models = [gr.Textbox(m, visible=False) for m in default_models]
 
     with gr.Column(scale=2):