Update app.py

app.py

@@ -35,8 +35,6 @@ MAX_SEED = np.iinfo(np.int32).max
 USE_TORCH_COMPILE = 0
 ENABLE_CPU_OFFLOAD = 0
 
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-
 
 if torch.cuda.is_available():
     pipe = StableDiffusionXLPipeline.from_pretrained(
@@ -45,10 +43,12 @@ if torch.cuda.is_available():
         use_safetensors=True,
     )
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-    pipe.to(device) 
+    
     
     pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
     pipe.set_adapters("dalle")
+
+    pipe.to("cuda")