Yaron Koresh commited on
Commit
bb96d35
·
verified ·
1 Parent(s): af4cba1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ import numpy as np
10
  from lxml.html import fromstring
11
  from diffusers import AutoPipelineForText2Image
12
  #from tempfile import NamedTemporaryFile
13
- from pathos.multiprocessing import Pool
14
 
15
  #model_id = "runwayml/stable-diffusion-v1-5"
16
  #model_id = "kandinsky-community/kandinsky-3"
@@ -193,7 +193,7 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
193
  def ret(idx):
194
  result[idx] = infer(prompt,prompt2)
195
  def rets():
196
- Pool().map(ret,[[i] for i in range(12)])
197
  run_button.click(rets)
198
 
199
  demo.queue().launch()
 
10
  from lxml.html import fromstring
11
  from diffusers import AutoPipelineForText2Image
12
  #from tempfile import NamedTemporaryFile
13
+ from pathos.multiprocessing import ProcessingPool as Pool
14
 
15
  #model_id = "runwayml/stable-diffusion-v1-5"
16
  #model_id = "kandinsky-community/kandinsky-3"
 
193
  def ret(idx):
194
  result[idx] = infer(prompt,prompt2)
195
  def rets():
196
+ Pool().map(ret,[str(i) for i in range(12)])
197
  run_button.click(rets)
198
 
199
  demo.queue().launch()