Update app.py

app.py

@@ -24,13 +24,12 @@ 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"
+    model = "tkurtulus/sdxl_tk"
     pipe = DiffusionPipeline.from_pretrained(
         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',