Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 from diffusers import StableDiffusionPipeline
 import os
 from huggingface_hub import login
+from safetensors.torch import load_file
 
 # Login using the token
 login(token=os.environ.get("HF_TOKEN"))
@@ -15,16 +16,10 @@ pipe = StableDiffusionPipeline.from_pretrained(
     use_auth_token=os.environ.get("HF_TOKEN")
 )
 
-# Load your LoRA weights
-lora_path = "naonauno/40k-half-sd15"
-pipe.load_lora_weights(
-    lora_path,
-    weight_name="40kHalf.safetensors",
-    use_auth_token=os.environ.get("HF_TOKEN")
-)
-
-# Set the cross attention processor scale
-pipe.fuse_lora(scale=1.0)
+# Load your LoRA weights from local file
+lora_path = "models/40kHalf.safetensors"  # Update this path to where you upload the file in your Space
+state_dict = load_file(lora_path)
+pipe.unet.load_attn_procs(state_dict)
 
 def generate_image(prompt, negative_prompt, guidance_scale, steps):
     with torch.no_grad():