Update app.py

app.py

@@ -10,7 +10,7 @@ import numpy as np
 from lxml.html import fromstring
 from diffusers import AutoPipelineForText2Image
 #from tempfile import NamedTemporaryFile
-from pathos.multiprocessing import ProcessingPool as Pool
+from pathos.multiprocessing import ThreadPool as Pool
 
 #model_id = "runwayml/stable-diffusion-v1-5"
 #model_id = "kandinsky-community/kandinsky-3"
@@ -190,10 +190,12 @@ 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):
+    def _ret(idx):
         result[idx] = infer(prompt,prompt2)
-    def rets():
-        Pool().map(ret,[str(i) for i in range(12)])
-    run_button.click(rets)
+    def _rets():
+        ln = len(result)
+        idxs = list(range(ln))
+        Pool(ln).amap( _ret, idxs )
+    run_button.click(_rets)
 
 demo.queue().launch()