Update app.py

app.py

@@ -28,7 +28,7 @@ state_dict = load_state_dict(model_file)
 model, _, _, _, _ = ControlNetModel_Union._load_pretrained_model(
     controlnet_model, state_dict, model_file, "xinsir/controlnet-union-sdxl-1.0"
 )
-device = torch.cuda.is_available()
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 model.to(device=device, dtype=torch.float16 if device == 'cuda' else torch.float32)
 vae = AutoencoderKL.from_pretrained(