Update app.py

app.py

@@ -14,13 +14,13 @@ import warnings
 import time
 import asyncio
 from functools import partial
+from concurrent.futures import ProcessPoolExecutor as Pool
 
 # external
 
 import spaces
 import torch
 import gradio as gr
-from pathos.multiprocessing import ProcessPool as Pool
 from numpy import asarray as array
 from lxml.html import fromstring
 from diffusers.utils import export_to_gif, load_image
@@ -136,7 +136,7 @@ pipe_flux = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_d
 
 def parallel(func,*args):
     with Pool(nodes=len(args)) as pool:
-        res = pool.imap(func, *args)
+        res = pool.map(func, *args)
     return list(res)
         
 # functionality