Update app.py

app.py

@@ -10,8 +10,9 @@ import numpy as np
 from lxml.html import fromstring
 #from transformers import pipeline
 from torch import multiprocessing as mp
+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
@@ -35,10 +36,6 @@ def init_pool(_1,_2):
     infer2 = _2
     port_inc()
 
-#pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to(device)
-#pipe2 = StableDiffusionXLImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to(device)
-#pipe2.unet = torch.compile(pipe2.unet, mode="reduce-overhead", fullgraph=True)
-
 def pipe_t2i():
     PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token"), device=-1)
     return PIPE
@@ -214,7 +211,7 @@ def main():
         return list(out)
     port_inc()
     
-    #mp.set_start_method("spawn", force=True)
+    mp.set_start_method("spawn", force=True)
 
     with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
         with gr.Column(elem_id="col-container"):