Update app.py

app.py

@@ -10,9 +10,9 @@ import numpy as np
 from lxml.html import fromstring
 #from transformers import pipeline
 from torch import multiprocessing as mp
-from torch.multiprocessing import Pool as Pool
+#from torch.multiprocessing import Pool as Pool
 #from pathos.multiprocessing import ProcessPool as Pool
-#from pathos.threading import ThreadPool as Pool
+from pathos.threading import ThreadPool as Pool
 #from diffusers.pipelines.flux import FluxPipeline
 #from diffusers.utils import export_to_gif
 #from huggingface_hub import hf_hub_download
@@ -181,7 +181,7 @@ def run(p1,p2,*result):
         rng = list(range(ln))
     
         arr = [p for _ in rng]
-        pool = Pool(ln)
+        pool = Pool()
         out = list(pool.imap(infer,arr))
         pool.close()
         pool.join()