Update app.py

app.py

@@ -37,8 +37,7 @@ vae = AutoencoderKL.from_pretrained(
 )
 
 # Ensure model and scheduler are initialized in GPU-enabled function
-if torch.cuda.is_available():
-    pipe = StableDiffusionXLPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16).to(device)
+pipe = StableDiffusionXLPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16).to(device)
 
 pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)