Update app.py

app.py

@@ -29,10 +29,6 @@ def port_inc():
     else:
         os.environ["CUSTOM_PORT"]=str(int(env)+1)
 
-def init_pool():
-    #port_inc()
-    print("init pool")
-
 def pipe_t2i():
     PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to("cuda")
     return PIPE
@@ -193,9 +189,9 @@ def run(p1,p2,*result):
         arr = [p for _ in rng]
         out = None
         print("Starting pool!")
-        with Pool(ln, initializer=init_pool) as pool:
-            out = pool.imap(infer,arr)
-        return list(out)
+        with Pool(ln) as pool:
+            out = pool.map(infer,arr)
+        return list(out.get())
 
 def ui():
     with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo: