Update app.py

app.py

@@ -94,7 +94,7 @@ pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
 pipe.load_lora_weights("briaai/BRIA-2.3-FAST-LORA")
 pipe.fuse_lora()
 
-pipe = pipe.to('cuda:0')
+pipe = pipe.to(device)
 # pipe.enable_xformers_memory_efficient_attention()
 
 generator = torch.Generator(device='cuda:0').manual_seed(123456)