Update app.py

app.py

@@ -40,16 +40,12 @@ def init_pool(_1,_2):
 #pipe2.unet = torch.compile(pipe2.unet, mode="reduce-overhead", fullgraph=True)
 
 def pipe_t2i():
-    global PIPE
-    if PIPE is None:
-        PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token"), device=-1)
+    PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token"), device=-1)
     return PIPE
     
 def pipe_i2i():
-    global PIPE
-    if PIPE is None:
-        PIPE = pipeline("image-to-image", model="stabilityai/stable-diffusion-xl-refiner-1.0", tokenizer="stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, device=-1, variant="fp16", use_safetensors=True)
-        PIPE.unet = torch.compile(PIPE.unet, mode="reduce-overhead", fullgraph=True)
+    PIPE = pipeline("image-to-image", model="stabilityai/stable-diffusion-xl-refiner-1.0", tokenizer="stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, device=-1, variant="fp16", use_safetensors=True)
+    PIPE.unet = torch.compile(PIPE.unet, mode="reduce-overhead", fullgraph=True)
     return PIPE
 
 pp1=pipe_t2i()