Update app.py

app.py

@@ -6,9 +6,10 @@ from diffusers import StableDiffusionXLPipeline
 from diffusers import AutoencoderTiny, AutoencoderKL
 
 dtype = torch.bfloat16
+device = "cuda" if torch.cuda.is_available() else "cpu"
 #taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
 good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=dtype).to(device)
-pipeline = StableDiffusionXLPipeline.from_pretrained("bobber/bigasp2", torch_dtype=dtype, vae=good_vae).to("cuda")
+pipeline = StableDiffusionXLPipeline.from_pretrained("bobber/bigasp2", torch_dtype=dtype, vae=good_vae).to(device)
 
 @spaces.GPU
 def generate(prompt, negative_prompt, width, height, sample_steps, guidance_scale):