Update app.py

app.py

@@ -136,19 +136,10 @@ pipe_flux = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_d
 
 # Parallelism
 
-def parallel(*pairs):
-    if len(pairs) == 0:
-        return
-    if len(pairs) == 1:
-        pairs = pairs[0]
-    running_tasks = []
-    for pair in pairs:
-        task = pair.pop(0)
-        running_tasks.append(Process(target=task, args=(pair,)))
-    for running_task in running_tasks:
-        running_task.start()
-    for running_task in running_tasks:
-        running_task.join()
+def parallel(workers,func,*args):
+    with ProcessPoolExecutor(workers) as ex:
+        res = ex.map(func, args)
+    return list(res)
         
 # functionality
         
@@ -241,11 +232,6 @@ def pipe_generate(img,p1,p2,motion):
 
 def handle_generate(*inp):
 
-    calc_out = []
-    
-    def calc(*args):
-        calc_out.append(pipe_generate(*args))
-
     inp = list(inp)
     
     inp[1] = translate(inp[1],"english")
@@ -263,14 +249,12 @@ def handle_generate(*inp):
     
     ln = len(result)
 
-    parargs = [[calc,*inp] for i in range(ln)]
-    parallel(parargs)
+    pipe_out = parallel(ln,pipe_generate,*inp)
     names = []
-    for i in calc_out:
+    for i in pipe_out:
         name = generate_random_string(12)+".png"
         export_to_gif(i.frames[0],name,fps=fps)
         names.append( name )
-    
     return names
 
 def ui():