Update app.py

app.py

@@ -10,29 +10,30 @@ theme = gr.themes.Base(
     font=[gr.themes.GoogleFont('Libre Franklin'), gr.themes.GoogleFont('Public Sans'), 'system-ui', 'sans-serif'],
 )
 
+device="cuda"
 pipe_xlc = AutoPipelineForText2Image.from_pretrained(
     "temp-org-cc/CommonCanvas-XLC",
     custom_pipeline="multimodalart/sdxl_perturbed_attention_guidance",
     torch_dtype=torch.float16
-)
+).to(device)
+
 pipe_xlnc = AutoPipelineForText2Image.from_pretrained(
     "temp-org-cc/CommonCanvas-XLNC",
     custom_pipeline="multimodalart/sdxl_perturbed_attention_guidance",
     torch_dtype=torch.float16
-)
+).to(device)
+
 pipe_sc = AutoPipelineForText2Image.from_pretrained(
     "temp-org-cc/CommonCanvas-SC",
     custom_pipeline="hyoungwoncho/sd_perturbed_attention_guidance",
     torch_dtype=torch.float16
-)
+).to(device)
+
 pipe_snc = AutoPipelineForText2Image.from_pretrained(
     "temp-org-cc/CommonCanvas-SNC",
     custom_pipeline="hyoungwoncho/sd_perturbed_attention_guidance",
     torch_dtype=torch.float16
-)
-
-device="cuda"
-pipe = pipe.to(device)
+).to(device)
 
 @spaces.GPU
 def run_xlc(prompt, negative_prompt=None, guidance_scale=7.0, pag_scale=3.0, pag_layers=["mid"], randomize_seed=True, seed=42, progress=gr.Progress(track_tqdm=True)):