update

app.py

@@ -37,15 +37,6 @@ device = devicetorch.get(torch)
 #        "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, variant="fp16"
 #    ).to("cuda")
 
-pipe_sdxl = StableDiffusionXLPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
-    torch_dtype=torch.float16,
-    variant="fp16",
-).to(device)
-pipe_sd15 = StableDiffusionPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, variant="fp16"
-).to(device)
-
 #@spaces.GPU(enable_queue=True)
 def generate_image(
     prompt,
@@ -57,6 +48,16 @@ def generate_image(
     global loaded
     checkpoint = checkpoints[ckpt][0].format(mode)
     guidance_scale = checkpoints[ckpt][2]
+
+    pipe_sdxl = StableDiffusionXLPipeline.from_pretrained(
+        "stabilityai/stable-diffusion-xl-base-1.0",
+        torch_dtype=torch.float16,
+        variant="fp16",
+    ).to(device)
+    pipe_sd15 = StableDiffusionPipeline.from_pretrained(
+        "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, variant="fp16"
+    ).to(device)
+
     pipe = pipe_sdxl if mode == "sdxl" else pipe_sd15
 
     if loaded != (ckpt + mode):