Update app.py

app.py

@@ -24,11 +24,14 @@ ENABLE_CPU_OFFLOAD = os.getenv('ENABLE_CPU_OFFLOAD') == '1'
 
 device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 if torch.cuda.is_available():
+    model = "stabilityai/stable-diffusion-xl-base-1.0"
     pipe = DiffusionPipeline.from_pretrained(
-        'stabilityai/stable-diffusion-xl-base-1.0',
+        model,
         torch_dtype=torch.float16,
         use_safetensors=True,
         variant='fp16')
+    pipe.load_lora_weights("model", weight_name="tk_pytorch_lora_weights.safetensors")
+    
     refiner = DiffusionPipeline.from_pretrained(
         'stabilityai/stable-diffusion-xl-refiner-1.0',
         torch_dtype=torch.float16,