Update app.py

app.py

@@ -4,9 +4,11 @@ import spaces
 import torch
 from diffusers import StableDiffusionXLPipeline
 
-pipeline = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0").to("cuda")
+pipeline = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
 pipeline.load_lora_weights("nroggendorff/zelda-lora")
 
+pipeline.to("cuda")
+
 @spaces.GPU
 def generate(prompt, negative_prompt, width, height, sample_steps):
     return pipeline(prompt=prompt, negative_prompt=negative_prompt, width=width, height=height, num_inference_steps=sample_steps).images[0]