Update app.py

app.py

@@ -9,9 +9,9 @@ import gradio as gr
 import numpy as np
 from lxml.html import fromstring
 from transformers import pipeline
-from torch import multiprocessing as mp, _dynamo
+#from torch.multiprocessing import 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
@@ -138,9 +138,10 @@ def main(p1,p2,*result):
     ln = len(result)
     rng = list(range(ln))
     arr = [p for _ in rng]
-    with mp.Pool(ln, initializer=init_pool) as pool:
+    out = None
+    with Pool(ln, initializer=init_pool) as pool:
         out = pool.imap(infer,arr)
-        return list(out)
+    return list(out)
 
 def infer(p):
     p1 = p["a"]