Update app.py

app.py

@@ -90,10 +90,7 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
     if task.done() and result is not None and not isinstance(result, tuple):
         with lock:
             png_path =  model_str.replace("/", "_") + " - " + get_current_time() + "_" + str(theSeed) + ".png"
-            # lnk_output(value=png_path)
-            
             image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, theSeed)
-            imgageHasUpdated(result)
         return image
     return None
 
@@ -115,6 +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(image)
     return gallery
 
 JS="""
@@ -228,6 +226,7 @@ with gr.Blocks(head=CSS + JS) as demo:
     for m, o in zip(current_models, output):
         gen_event = gr.on(triggers=[gen_button.click, txt_input.submit], fn=gen_fn,inputs=[m, txt_input, neg_input, height, width, steps, cfg, seed], outputs=[o], concurrency_limit=None, queue=False)
         o.change(add_gallery, [o, m, gallery], [gallery])
+        # o.change(imgageHasUpdated,[o])
 
     with gr.Column(scale=4):
         with gr.Accordion('Model selection'):