Update app.py

app.py

@@ -132,13 +132,6 @@ pipe.enable_free_init(method="butterworth", use_fast_sampling=fast)
 
 pipe_flux = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to(device,dtype=dtype)
 
-# Parallelism
-
-def parallel(func,*args):
-    with Pool(len(args)) as pool:
-        res = pool.map(func, args)
-    return list(res)
-        
 # functionality
         
 def run(cmd):
@@ -246,9 +239,8 @@ def handle_generate(*inp):
     inp[1] = ", ".join(_do)
     
     ln = len(result)
-    parallel_args = [inp for i in range(ln)]
+    pipe_out = [pipe_generate(*inp) for i in range(ln)]
     
-    pipe_out = parallel( pipe_generate, *parallel_args )
     names = []
     for i in pipe_out:
         name = generate_random_string(12)+".png"