Update app.py

app.py

@@ -23,7 +23,7 @@ base_model="stabilityai/stable-diffusion-xl-base-1.0"
 # ).to("cuda")
 base = DiffusionPipeline.from_pretrained(base_model,
         torch_dtype=torch.float16, use_safetensors=True
-).to("cuda")
+).to
 lora_model_path = "./pytorch_weights.safetensors"
 
 base.load_lora_weights(lora_model_path)