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
+from torch.multiprocessing import Pool2
 #from pathos.multiprocessing import ProcessPool as Pool
-from pathos.threading import ThreadPool as Pool
+from pathos.threading import ThreadPool as Pool1
 #from diffusers.pipelines.flux import FluxPipeline
 #from diffusers.utils import export_to_gif
 #from huggingface_hub import hf_hub_download
@@ -189,15 +189,21 @@ def run(p1,p2,*result):
         ln = len(result)
         print("images: "+str(ln))
         rng = list(range(ln))
+    
         arr1 = [p for _ in rng]
-        print("Starting pool!")
-        pool = Pool(nodes=ln*2)
-        out1 = list(pool.imap(infer1,arr1))
+        pool1 = Pool1(ln)
+        out1 = list(pool1.imap(infer1,arr1))
+        pool1.close()
+        pool1.join()
+        pool1.clear()
+    
         arr2 = [{"a":p1_en,"b":p2_en,"c":out1[_]} for _ in rng]
-        out2 = list(pool.imap(infer2,arr2))
-        pool.close()
-        pool.join()
-        pool.clear()
+        pool2 = Pool2(ln)
+        out2 = list(pool2.imap(infer2,arr2))
+        pool2.close()
+        pool2.join()
+        pool2.clear()
+    
         return out2
 
 def ui():