Update app.py

app.py

@@ -190,17 +190,15 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
             result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
             result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
 
-    def _ret(idx):
-        p1 = prompt.value
-        p2 = prompt2.value
-        print(f'Starting {idx}: {result[idx]} {p1} {p2}')
+    def _ret(idx,p1,p2):
+        print(f'Starting {idx}: {p1} {p2}')
         v = infer(p1,p2)
         print(f'Finished {idx}: {v}')
-        result[idx].value=v
-    def _rets():
+        return v
+    def _rets(p1,p2):
         ln = len(result)
-        idxs = list(range(ln))
-        Pool(ln).imap( _ret, idxs )
-    run_button.click(fn=_rets)
+        idxs = [ [idx,p1,p2] for idx in list(range(ln)) ]
+        return Pool(ln).imap( _ret, idxs ).get()
+    run_button.click(fn=_rets,inputs=[prompt,prompt2],outputs=result)
 
 demo.queue().launch()