Update app.py

app.py

@@ -10,10 +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 ProcessPoolExecutor
-
-#pool = ProcessPoolExecutor(1000)
-#pool.__enter__()
+from pathos.multiprocessing import Pool
 
 #model_id = "runwayml/stable-diffusion-v1-5"
 #model_id = "kandinsky-community/kandinsky-3"
@@ -93,8 +90,8 @@ def Piper(_do,_dont):
         height=320,
         width=576,
         negative_prompt=_dont,
-        num_inference_steps=100,
-        guidance_scale=10
+        num_inference_steps=50,
+        guidance_scale=2
     )
 
 def infer(prompt,prompt2):
@@ -132,7 +129,7 @@ footer {
     margin: 0 auto !important;
     max-width: 15cm !important;
  }
-#image-container {
+.image-container {
     aspect-ratio: 576 / 320 !important;
 }
 .dropdown-arrow {
@@ -175,21 +172,23 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
             )
         with gr.Row():
             run_button = gr.Button("התחלה")
-        result = gr.Image(elem_id="image-container", label="Result", show_label=False, type='filepath', show_share_button=False)
-    prompt.submit(
-        fn = infer,
-        inputs = [prompt,prompt2],
-        outputs = [result]
-    )
-    prompt2.submit(
-        fn = infer,
-        inputs = [prompt,prompt2],
-        outputs = [result]
-    )
-    run_button.click(
-        fn = infer,
-        inputs = [prompt,prompt2],
-        outputs = [result]
-    )
+        with gr.Row():
+            result[0] = gr.Image(elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False)
+            result[1] = gr.Image(elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False)
+            result[2] = gr.Image(elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False)
+        with gr.Row():
+            result[3] = gr.Image(elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False)
+            result[4] = gr.Image(elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False)
+            result[5] = gr.Image(elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False)
+        with gr.Row():
+            result[6] = gr.Image(elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False)
+            result[7] = gr.Image(elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False)
+            result[8] = gr.Image(elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False)
+
+    def ret(idx):
+        result[idx] = infer(prompt,prompt2)
+    def rets():
+        Pool().map(ret,[range(9)])
+    run_button.click(rets)
 
 demo.queue().launch()